ford442 commited on
Commit
7bae417
·
verified ·
1 Parent(s): 2298f8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -86,11 +86,11 @@ pipe.to(device=device, dtype=torch.bfloat16)
86
  #pipe.to(device)
87
  pipe.vae=vaeX.to('cpu')
88
 
89
- text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(device=device, dtype=torch.bfloat16)
90
- text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(device=device, dtype=torch.bfloat16)
91
- text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True).to(device=device, dtype=torch.bfloat16)
92
 
93
- upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
94
 
95
  MAX_SEED = np.iinfo(np.int32).max
96
 
 
86
  #pipe.to(device)
87
  pipe.vae=vaeX.to('cpu')
88
 
89
+ text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(device)
90
+ text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(device)
91
+ text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True).to(device)
92
 
93
+ upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to('cpu') #.to(device)
94
 
95
  MAX_SEED = np.iinfo(np.int32).max
96