Update model.py

model.py

@@ -196,10 +196,10 @@ class Model:
         image = resize_image_to_16_multiple(image, 'depth')
         W, H = image.size
         print(W, H)
-        # self.t5_model.model.to(self.device)
-        # self.gpt_model_depth.to(self.device)
-        # self.get_control_depth.model.to(self.device)
-        # self.vq_model.to(self.device)
+        self.t5_model.model.to(self.device)
+        self.gpt_model_depth.to(self.device)
+        self.get_control_depth.model.to(self.device)
+        self.vq_model.to(self.device)
         image_tensor = torch.from_numpy(np.array(image)).to(self.device)
         # condition_img = torch.from_numpy(
         #     self.get_control_depth(image_tensor)).unsqueeze(0)