Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,9 @@ with open('loras.json', 'r') as f:
|
|
| 47 |
|
| 48 |
#pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-uint4-svd-r32", torch_dtype=torch.bfloat16)
|
| 49 |
#torch.cuda.empty_cache()
|
| 50 |
-
pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF16", torch_dtype=torch.bfloat16)
|
|
|
|
|
|
|
| 51 |
|
| 52 |
#pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
|
| 53 |
pipe.vae = AutoencoderKL.from_pretrained("AlekseyCalvin/Custom_VAE-Z-image-FLUX.1-by-G-REPA", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
|
|
| 47 |
|
| 48 |
#pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-uint4-svd-r32", torch_dtype=torch.bfloat16)
|
| 49 |
#torch.cuda.empty_cache()
|
| 50 |
+
#pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF16", torch_dtype=torch.bfloat16)
|
| 51 |
+
pipe = diffusers.ZImagePipeline.from_pretrained("AlekseyCalvin/Z-Image-Deturbo-Returbo-Base_Diffusers", torch_dtype=torch.bfloat16)
|
| 52 |
+
|
| 53 |
|
| 54 |
#pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
|
| 55 |
pipe.vae = AutoencoderKL.from_pretrained("AlekseyCalvin/Custom_VAE-Z-image-FLUX.1-by-G-REPA", torch_dtype=torch.bfloat16, device_map="cuda")
|