Spaces:

prokofyev8
/

Z-Image-Re-Turbo

Runtime error

prokofyev8 commited on 13 days ago

Commit

0a1ab5c

verified ·

1 Parent(s): 8c2e3d4

changed to "ostris/Z-Image-De-Turbo"

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import random
 from diffusers import DiffusionPipeline
 import torch
-pipe = DiffusionPipeline.from_pretrained("GuangyuanSD/Z-Image-Re-Turbo-LoRA", dtype=torch.bfloat16, device_map="cuda")
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024

 from diffusers import DiffusionPipeline
 import torch
+pipe = DiffusionPipeline.from_pretrained("ostris/Z-Image-De-Turbo", dtype=torch.bfloat16, device_map="cuda")
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024