Update app.py

app.py

@@ -44,8 +44,9 @@ with open('loras.json', 'r') as f:
 #pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
 ##pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 
-pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-uint4-svd-r32", torch_dtype=torch.bfloat16)
-torch.cuda.empty_cache()
+#pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-uint4-svd-r32", torch_dtype=torch.bfloat16)
+#torch.cuda.empty_cache()
+pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF16", torch_dtype=torch.bfloat16)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"