ford442 commited on
Commit
4a877eb
·
verified ·
1 Parent(s): 1315b05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ checkpoint = "microsoft/Phi-3.5-mini-instruct"
65
  pipe = StableDiffusion3Pipeline.from_pretrained(
66
  "stabilityai/stable-diffusion-3.5-large",
67
  #"ford442/stable-diffusion-3.5-large-bf16",
68
- vae=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='vae',token=True),
69
  transformer=SD3Transformer2DModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='transformer',token=True),
70
  text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder',token=True),
71
  text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True),
 
65
  pipe = StableDiffusion3Pipeline.from_pretrained(
66
  "stabilityai/stable-diffusion-3.5-large",
67
  #"ford442/stable-diffusion-3.5-large-bf16",
68
+ vae=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", use_safetensors=True, subfolder='vae',token=True),
69
  transformer=SD3Transformer2DModel.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='transformer',token=True),
70
  text_encoder=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder',token=True),
71
  text_encoder_2=CLIPTextModelWithProjection.from_pretrained("ford442/stable-diffusion-3.5-large-bf16", subfolder='text_encoder_2',token=True),