init

app.py

@@ -363,7 +363,9 @@ def create_demo(model):
                            
 def main():
     model = ControlNetDepthDesignModelMulti()
+    model = model.to(device)
     print('Models uploaded successfully')
+    print(model.device)
     
     title = "# StableDesign"
     description = """