primerz commited on
Commit
6db28b5
·
verified ·
1 Parent(s): fd6d0ea

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -152,7 +152,7 @@ def load_controlnets():
152
  """Load ControlNet models."""
153
  print("Loading ControlNet Zoe Depth model...")
154
  controlnet_depth = ControlNetModel.from_pretrained(
155
- "xinsir/controlnet-openpose-sdxl-1.0",
156
  torch_dtype=dtype
157
  ).to(device)
158
  print(" [OK] ControlNet Depth loaded")
 
152
  """Load ControlNet models."""
153
  print("Loading ControlNet Zoe Depth model...")
154
  controlnet_depth = ControlNetModel.from_pretrained(
155
+ "xinsir/controlnet-depth-sdxl-1.0",
156
  torch_dtype=dtype
157
  ).to(device)
158
  print(" [OK] ControlNet Depth loaded")