remove depth

model.py

@@ -50,9 +50,9 @@ class Model:
         self.vq_model = self.load_vq()
         self.t5_model = self.load_t5()
         self.gpt_model_canny = self.load_gpt(condition_type='canny')
-        self.gpt_model_depth = self.load_gpt(condition_type='depth')
+        # self.gpt_model_depth = self.load_gpt(condition_type='depth')
         self.get_control_canny = CannyDetector()
-        self.get_control_depth = MidasDetector('cuda')
+        # self.get_control_depth = MidasDetector('cuda')
         # self.get_control_depth =  MidasDetector.from_pretrained("lllyasviel/Annotators")
 
     def to(self, device):
@@ -123,7 +123,7 @@ class Model:
         image = resize_image_to_16_multiple(image, 'canny')
         W, H = image.size
         print(W, H)
-        self.gpt_model_depth.to('cpu')
+        # self.gpt_model_depth.to('cpu')
         self.t5_model.model.to('cuda')
         self.gpt_model_canny.to('cuda')
         self.vq_model.to('cuda')