linoyts HF Staff commited on
Commit
ec6a7bd
·
verified ·
1 Parent(s): a063df3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -11,11 +11,12 @@ pipe = ZImagePipeline.from_pretrained(
11
  low_cpu_mem_usage=False,
12
  )
13
  pipe.load_lora_weights("bdsqlsz/qinglong_DetailedEyes_Z-Image", weight_name="qinglong_detailedeye_z-imageV2(comfy).safetensors")
 
14
  pipe.to("cuda")
15
 
16
  # ======== AoTI compilation + FA3 ========
17
- pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
18
- spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
19
 
20
  print("Pipeline loaded!")
21
 
 
11
  low_cpu_mem_usage=False,
12
  )
13
  pipe.load_lora_weights("bdsqlsz/qinglong_DetailedEyes_Z-Image", weight_name="qinglong_detailedeye_z-imageV2(comfy).safetensors")
14
+
15
  pipe.to("cuda")
16
 
17
  # ======== AoTI compilation + FA3 ========
18
+ # pipe.transformer.layers._repeated_blocks = ["ZImageTransformerBlock"]
19
+ # spaces.aoti_blocks_load(pipe.transformer.layers, "zerogpu-aoti/Z-Image", variant="fa3")
20
 
21
  print("Pipeline loaded!")
22