Spaces:

credent007
/

easyocr-phi3

Paused

credent007 commited on Apr 10

Commit

0ebb85e

verified ·

1 Parent(s): c3b483d

Update llm.py

Files changed (1) hide show

llm.py CHANGED Viewed

@@ -17,7 +17,7 @@ processor = AutoProcessor.from_pretrained(
 model = AutoModelForImageTextToText.from_pretrained(
     model_name,
     device_map="auto",
-    torch_dtype=torch.float16,
     trust_remote_code=True
 )

 model = AutoModelForImageTextToText.from_pretrained(
     model_name,
     device_map="auto",
+    dtype=torch.float16,
     trust_remote_code=True
 )