init

app.py

@@ -226,8 +226,16 @@ class ControlNetDepthDesignModelMulti:
                            If the size is not the same the submission will fail.
         """
         print(prompt)
+        if torch.cuda.is_available():
+            # Print the number of available GPUs
+            print("Available GPU devices:")
+            for i in range(torch.cuda.device_count()):
+                print(f"Device {i}: {torch.cuda.get_device_name(i)}")
+        else:
+            print("No GPU devices available. Using CPU.")
+    
         print(self.depth_estimator.device)
-        print(self.image_segmentor.device)
+        print(self.pipe.device)
         flush()
         self.generator = torch.Generator(device=device).manual_seed(self.seed)