AlekseyCalvin commited on
Commit
78092b2
·
verified ·
1 Parent(s): d1c7fd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -53,8 +53,8 @@ with open('loras.json', 'r') as f:
53
  #pipe = diffusers.ZImagePipeline.from_pretrained("AlekseyCalvin/Z_Image_Deturbo_Diffusers", torch_dtype=torch.bfloat16)
54
  pipeline = 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")
58
 
59
 
60
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 
53
  #pipe = diffusers.ZImagePipeline.from_pretrained("AlekseyCalvin/Z_Image_Deturbo_Diffusers", torch_dtype=torch.bfloat16)
54
  pipeline = 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")
58
 
59
 
60
  #pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")