Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ with open('loras.json', 'r') as f:
|
|
| 51 |
#torch.cuda.empty_cache()
|
| 52 |
#pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF16", torch_dtype=torch.bfloat16)
|
| 53 |
#pipe = diffusers.ZImagePipeline.from_pretrained("AlekseyCalvin/Z_Image_Deturbo_Diffusers", torch_dtype=torch.bfloat16)
|
| 54 |
-
|
| 55 |
|
| 56 |
#pipe.text_encoder = Qwen3ForCausalLM.from_pretrained('Qwen/Qwen3-4B-Instruct-2507').to(torch.bfloat16)
|
| 57 |
#pipe.vae = AutoencoderKL.from_pretrained("AlekseyCalvin/Custom_VAE-Z-image-FLUX.1-by-G-REPA", torch_dtype=torch.bfloat16, device_map="cuda")
|
|
|
|
| 51 |
#torch.cuda.empty_cache()
|
| 52 |
#pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF16", torch_dtype=torch.bfloat16)
|
| 53 |
#pipe = diffusers.ZImagePipeline.from_pretrained("AlekseyCalvin/Z_Image_Deturbo_Diffusers", torch_dtype=torch.bfloat16)
|
| 54 |
+
pipe = ZImagePipeline.from_pretrained('AlekseyCalvin/Z-Image-Deturbo-Returbo-Base_Diffusers', torch_dtype=torch.bfloat16)
|
| 55 |
|
| 56 |
#pipe.text_encoder = Qwen3ForCausalLM.from_pretrained('Qwen/Qwen3-4B-Instruct-2507').to(torch.bfloat16)
|
| 57 |
#pipe.vae = AutoencoderKL.from_pretrained("AlekseyCalvin/Custom_VAE-Z-image-FLUX.1-by-G-REPA", torch_dtype=torch.bfloat16, device_map="cuda")
|