Update model.py

model.py

@@ -197,12 +197,9 @@ class Model:
         control_strength: float,
         preprocessor_name: str
     ) -> list[PIL.Image.Image]:
-        # self.gpt_model_edge.to('cpu')
-        self.t5_model.model.to(self.device)
-        # self.gpt_model_depth.to(self.device)
+        self.t5_model.model.to(self.device).to(torch.bfloat16)
         self.load_gpt_weight('depth')
         self.gpt_model.to('cuda').to(torch.bfloat16)
-        # self.get_control_depth.model.to(self.device)
         self.vq_model.to(self.device)
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image)