Z-Image-Turbo

Running on Zero

rahul7star commited on 18 days ago

Commit

78bec1d

verified ·

1 Parent(s): 8802f81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ print("Loading Z-Image-Turbo pipeline...")
 pipe = DiffusionPipeline.from_pretrained(
     "Tongyi-MAI/Z-Image-Turbo",#"T5B/Z-Image-Turbo-FP8",
     torch_dtype=torch.bfloat16,
-    low_cpu_mem_usage=True,
 )
 pipe.to("cuda")
 print("Pipeline loaded!")

 pipe = DiffusionPipeline.from_pretrained(
     "Tongyi-MAI/Z-Image-Turbo",#"T5B/Z-Image-Turbo-FP8",
     torch_dtype=torch.bfloat16,
+    low_cpu_mem_usage=False,
 )
 pipe.to("cuda")
 print("Pipeline loaded!")