init

app.py

@@ -180,7 +180,7 @@ class ControlNetDepthDesignModelMulti:
     """ Produces random noise images """
     def __init__(self):
         """ Initialize your model(s) here """
-
+        device = torch.device("cuda")
         #os.environ['HF_HUB_OFFLINE'] = "True"
         controlnet_depth= ControlNetModel.from_pretrained(
             "controlnet_depth", torch_dtype=dtype, use_safetensors=True)
@@ -363,9 +363,7 @@ def create_demo(model):
                            
 def main():
     model = ControlNetDepthDesignModelMulti()
-    model = model.to(device)
     print('Models uploaded successfully')
-    print(model.device)
     
     title = "# StableDesign"
     description = """