1inkusFace commited on
Commit
f863db1
·
verified ·
1 Parent(s): 87de474

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -81,9 +81,9 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
81
  #torch_dtype=torch.bfloat16,
82
  #use_safetensors=False,
83
  )
84
- text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', torch_dtype=torch.bfloat16, token=True).to(device)
85
- text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2', torch_dtype=torch.bfloat16,token=True).to(device)
86
- text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3', torch_dtype=torch.bfloat16,token=True).to(device)
87
 
88
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
89
 
 
81
  #torch_dtype=torch.bfloat16,
82
  #use_safetensors=False,
83
  )
84
+ text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(device=device, dtype=torch.bfloat16)
85
+ 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)
86
+ 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)
87
 
88
  pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")
89