update

model.py

@@ -191,6 +191,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)
         image_tensor = torch.from_numpy(np.array(image)).to(self.device)
         condition_img = torch.from_numpy(
             self.get_control_depth(image_tensor)).unsqueeze(0)