1inkusFace commited on
Commit
9e90391
·
verified ·
1 Parent(s): 6e178e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -137,7 +137,7 @@ def upload_to_ftp(filename):
137
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
138
 
139
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
140
- 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)
141
  pipe = StableDiffusion3Pipeline.from_pretrained(
142
  #"stabilityai # stable-diffusion-3.5-large",
143
  "ford442/stable-diffusion-3.5-large-bf16",
 
137
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
138
 
139
  #vae=AutoencoderKL.from_pretrained("ford442/sdxl-vae-bf16", use_safetensors=True, subfolder='vae',token=True)
140
+ 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)
141
  pipe = StableDiffusion3Pipeline.from_pretrained(
142
  #"stabilityai # stable-diffusion-3.5-large",
143
  "ford442/stable-diffusion-3.5-large-bf16",