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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ if torch.cuda.is_available():
33
  # load pipe
34
  controlnet = ControlNetModel.from_pretrained(
35
  controlnet_model_id,
36
- # variant="fp16",
37
  use_safetensors=True,
38
  torch_dtype=torch.float32
39
  )
@@ -42,7 +42,7 @@ if torch.cuda.is_available():
42
  base_model_id,
43
  controlnet=controlnet,
44
  vae=vae,
45
- # variant="fp16",
46
  use_safetensors=True,
47
  torch_dtype=torch.float32,
48
  )
 
33
  # load pipe
34
  controlnet = ControlNetModel.from_pretrained(
35
  controlnet_model_id,
36
+ variant="fp16",
37
  use_safetensors=True,
38
  torch_dtype=torch.float32
39
  )
 
42
  base_model_id,
43
  controlnet=controlnet,
44
  vae=vae,
45
+ variant="fp16",
46
  use_safetensors=True,
47
  torch_dtype=torch.float32,
48
  )