Spaces:
Runtime error
Runtime error
Update models.py
Browse files
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-
|
| 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")
|