rahul7star commited on
Commit
145fc8a
·
verified ·
1 Parent(s): 5989cf8

Update app_lora1.py

Browse files
Files changed (1) hide show
  1. 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
  # ============================================================