jiuface commited on
Commit
bd30ded
·
verified ·
1 Parent(s): 527d179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ if torch.cuda.is_available():
37
  use_safetensors=True,
38
  torch_dtype=torch.float32
39
  )
40
- vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float16)
41
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
42
  base_model_id,
43
  controlnet=controlnet,
 
37
  use_safetensors=True,
38
  torch_dtype=torch.float32
39
  )
40
+ vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float32)
41
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
42
  base_model_id,
43
  controlnet=controlnet,