Fix transformers version bugs

VikParuchuri · Jun 30, 2024 · b7e41aa · b7e41aa
1 parent 2968267
commit b7e41aa
Show file tree

Hide file tree

Showing 5 changed files with 480 additions and 461 deletions.
diff --git a/ocr_app.py b/ocr_app.py
@@ -1,3 +1,6 @@
+import os
+os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1" # For some reason, transformers decided to use .isin for a simple op, which is not supported on MPS
+
 import io
 
 import pandas as pd