dagloop5 commited on
Commit
1ed087b
·
verified ·
1 Parent(s): f63287e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -78,6 +78,8 @@ from ltx_pipelines.utils.media_io import decode_audio_from_file, encode_video
78
  from ltx_core.loader.primitives import LoraPathStrengthAndSDOps
79
  from ltx_core.loader.sd_ops import LTXV_LORA_COMFY_RENAMING_MAP
80
 
 
 
81
  print(f"[ATTN] Before patch: memory_efficient_attention={_attn_mod.memory_efficient_attention}")
82
  try:
83
  from xformers.ops import memory_efficient_attention as _mea
 
78
  from ltx_core.loader.primitives import LoraPathStrengthAndSDOps
79
  from ltx_core.loader.sd_ops import LTXV_LORA_COMFY_RENAMING_MAP
80
 
81
+ from ltx_core.model.transformer import attention as _attn_mod
82
+
83
  print(f"[ATTN] Before patch: memory_efficient_attention={_attn_mod.memory_efficient_attention}")
84
  try:
85
  from xformers.ops import memory_efficient_attention as _mea