Spaces:
Running
on
Zero
Running
on
Zero
Update app_lora1.py
Browse files- app_lora1.py +4 -0
app_lora1.py
CHANGED
|
@@ -54,6 +54,10 @@ except Exception as e:
|
|
| 54 |
log(traceback.format_exc())
|
| 55 |
pipe = None
|
| 56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
# ============================================================
|
| 58 |
# HELPERS
|
| 59 |
# ============================================================
|
|
|
|
| 54 |
log(traceback.format_exc())
|
| 55 |
pipe = None
|
| 56 |
|
| 57 |
+
|
| 58 |
+
# ======== AoTI compilation + FA3 ========
|
| 59 |
+
pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
|
| 60 |
+
spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
|
| 61 |
# ============================================================
|
| 62 |
# HELPERS
|
| 63 |
# ============================================================
|