Update models/trocr.py

models/trocr.py

@@ -16,7 +16,8 @@ class TrOCROCR(OCRModel):
         self.device = device or ("cuda" if torch.cuda.is_available() else "cpu")
 
         self.processor = AutoProcessor.from_pretrained(model_name)
-        self.model = AutoModelForVision2Seq.from_pretrained(model_name, torch_dtype=torch.float16)
+        self.model = AutoModelForVision2Seq.from_pretrained(model_name)
+        # self.model = AutoModelForVision2Seq.from_pretrained(model_name, torch_dtype=torch.float16)
         self.model.to(self.device)
         self.model.eval()