Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ from optimization import compile_transformer
|
|
| 6 |
from hub_utils import _push_compiled_graph_to_hub
|
| 7 |
from huggingface_hub import whoami
|
| 8 |
import time
|
| 9 |
-
from fa3 import
|
| 10 |
|
| 11 |
# --- Model Loading ---
|
| 12 |
dtype = torch.bfloat16
|
|
|
|
| 6 |
from hub_utils import _push_compiled_graph_to_hub
|
| 7 |
from huggingface_hub import whoami
|
| 8 |
import time
|
| 9 |
+
from fa3 import FlashFluxAttnProcessor3_0
|
| 10 |
|
| 11 |
# --- Model Loading ---
|
| 12 |
dtype = torch.bfloat16
|