Z-Image-Turbo

Running on Zero

rahul7star commited on 20 days ago

Commit

8802f81

verified ·

1 Parent(s): fef77da

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from diffusers import DiffusionPipeline
 # Load the pipeline once at startup
 print("Loading Z-Image-Turbo pipeline...")
 pipe = DiffusionPipeline.from_pretrained(
-    "T5B/Z-Image-Turbo-FP8",
     torch_dtype=torch.bfloat16,
     low_cpu_mem_usage=True,
 )

 # Load the pipeline once at startup
 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,
 )