pixagram-dev

Runtime error

primerz commited on Nov 1

Commit

fd6d0ea

verified ·

1 Parent(s): 70d7108

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -152,7 +152,7 @@ def load_controlnets():
     """Load ControlNet models."""
     print("Loading ControlNet Zoe Depth model...")
     controlnet_depth = ControlNetModel.from_pretrained(
-        "thibaud/controlnet-openpose-sdxl-1.0",
         torch_dtype=dtype
     ).to(device)
     print("  [OK] ControlNet Depth loaded")
@@ -161,7 +161,7 @@ def load_controlnets():
     print("Loading ControlNet OpenPose model...")
     try:
         controlnet_openpose = ControlNetModel.from_pretrained(
-            "thibaud/controlnet-openpose-sdxl-1.0",
             torch_dtype=dtype
         ).to(device)
         print("  [OK] ControlNet OpenPose loaded")

     """Load ControlNet models."""
     print("Loading ControlNet Zoe Depth model...")
     controlnet_depth = ControlNetModel.from_pretrained(
+        "xinsir/controlnet-openpose-sdxl-1.0",
         torch_dtype=dtype
     ).to(device)
     print("  [OK] ControlNet Depth loaded")
     print("Loading ControlNet OpenPose model...")
     try:
         controlnet_openpose = ControlNetModel.from_pretrained(
+            "xinsir/controlnet-openpose-sdxl-1.0",
             torch_dtype=dtype
         ).to(device)
         print("  [OK] ControlNet OpenPose loaded")