dagloop5 commited on
Commit
f638ada
·
verified ·
1 Parent(s): c963051

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ except Exception as e:
99
 
100
  def _patch_attention_for_kv_cache():
101
  """Patch Attention.forward to accept pre-projected K/V."""
102
- from ltx_core.model.transformer.attention import Attention
103
 
104
  _original_forward = Attention.forward
105
 
 
99
 
100
  def _patch_attention_for_kv_cache():
101
  """Patch Attention.forward to accept pre-projected K/V."""
102
+ from ltx_core.model.transformer.attention import Attention, apply_rotary_emb
103
 
104
  _original_forward = Attention.forward
105