Update model.py

model.py

@@ -74,8 +74,8 @@ class Model:
 
     def load_gpt(self, condition_type='canny'):
         gpt_ckpt = models[condition_type]
-        # precision = torch.bfloat16
-        precision = torch.float32
+        precision = torch.bfloat16
+        # precision = torch.float32
         latent_size = 768 // 16
         gpt_model = GPT_models["GPT-XL"](
             block_size=latent_size**2,
@@ -93,8 +93,8 @@ class Model:
         return gpt_model
 
     def load_t5(self):
-        # precision = torch.bfloat16
-        precision = torch.float32
+        precision = torch.bfloat16
+        # precision = torch.float32
         t5_model = T5Embedder(
             device=self.device,
             local_cache=True,