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
|