K00B404 commited on
Commit
0f417de
Β·
verified Β·
1 Parent(s): 9074c50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -38,6 +38,7 @@ controlnet = ControlNetModel.from_pretrained(controlnet_path, use_safetensors=Fa
38
  # load SDXL pipeline
39
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
40
  base_model_path,
 
41
  controlnet=controlnet,
42
  torch_dtype=torch.float16,
43
  add_watermarker=False,
 
38
  # load SDXL pipeline
39
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
40
  base_model_path,
41
+ variant='fp16',
42
  controlnet=controlnet,
43
  torch_dtype=torch.float16,
44
  add_watermarker=False,