init

app.py

@@ -112,9 +112,9 @@ def segment_image(
 
 def get_depth_pipeline():
     feature_extractor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf",
-                                                           torch_dtype=dtype)
+                                                           torch_dtype=dtype).to(device)
     depth_estimator = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf",
-                                                                  torch_dtype=dtype)
+                                                                  torch_dtype=dtype).to(device)
     return feature_extractor, depth_estimator
 
 
@@ -209,8 +209,7 @@ class ControlNetDepthDesignModelMulti:
         
         self.seg_image_processor, self.image_segmentor = get_segmentation_pipeline()
         self.depth_feature_extractor, self.depth_estimator = get_depth_pipeline()
-        self.depth_estimator = self.depth_estimator.to(device)
-    
+        
     @spaces.GPU
     def generate_design(self, empty_room_image: Image, prompt: str, guidance_scale: int = 10, num_steps: int = 50, strength: float =0.9, img_size: int = 640) -> Image:
         """