Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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 |
|