Spaces:
Running
on
Zero
Running
on
Zero
Alexander Bagus
commited on
Commit
·
a6316fc
1
Parent(s):
7ca8d69
22
Browse files
app.py
CHANGED
|
@@ -72,9 +72,8 @@ pipe.to("cuda", torch.bfloat16)
|
|
| 72 |
# pipe.to("cuda")
|
| 73 |
|
| 74 |
# ======== AoTI compilation + FA3 ========
|
| 75 |
-
pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
|
| 76 |
-
spaces.aoti_blocks_load(pipe.transformer.layers,
|
| 77 |
-
"zerogpu-aoti/Z-Image", variant="fa3")
|
| 78 |
|
| 79 |
def prepare(prompt):
|
| 80 |
polished_prompt = polish_prompt(prompt)
|
|
|
|
| 72 |
# pipe.to("cuda")
|
| 73 |
|
| 74 |
# ======== AoTI compilation + FA3 ========
|
| 75 |
+
# pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
|
| 76 |
+
# spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
|
|
|
|
| 77 |
|
| 78 |
def prepare(prompt):
|
| 79 |
polished_prompt = polish_prompt(prompt)
|