1inkusFace commited on
Commit
596417c
·
verified ·
1 Parent(s): 47b1816

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ pyx = cyper.inline(code, fast_indexing=True, directives=dict(boundscheck=False,
62
 
63
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
64
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
65
- vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, token=True)
66
  pipe = StableDiffusion3Pipeline.from_pretrained(
67
  #"stabilityai # stable-diffusion-3.5-large",
68
  "ford442/stable-diffusion-3.5-large-bf16",
 
62
 
63
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
64
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
65
+ vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
66
  pipe = StableDiffusion3Pipeline.from_pretrained(
67
  #"stabilityai # stable-diffusion-3.5-large",
68
  "ford442/stable-diffusion-3.5-large-bf16",