Spaces:

1inkusFace
/

StableDiffusion-3.5-Large-lora

Sleeping

1inkusFace commited on about 1 month ago

Commit

9300c66

verified ·

1 Parent(s): 6a35169

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,9 +85,9 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
     #torch_dtype=torch.bfloat16,
     #use_safetensors=False,
 )
-text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(torch.device("cuda:0", dtype=torch.bfloat16)
-text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(torch.device("cuda:0", dtype=torch.bfloat16)
-text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True).to(torch.device("cuda:0", dtype=torch.bfloat16)
 pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")

     #torch_dtype=torch.bfloat16,
     #use_safetensors=False,
 )
+text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder', token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)
+text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)
+text_encoder_3=T5EncoderModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_3',token=True).to(torch.device("cuda:0"), dtype=torch.bfloat16)
 pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safetensors")