Skip to content

Commit

Permalink
Merge pull request #15 from VikParuchuri/dev
Browse files Browse the repository at this point in the history
Fix transformers version bugs
  • Loading branch information
VikParuchuri authored Jun 30, 2024
2 parents 81b1bde + b7e41aa commit 2ddc278
Show file tree
Hide file tree
Showing 5 changed files with 480 additions and 461 deletions.
3 changes: 3 additions & 0 deletions ocr_app.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
Loading

0 comments on commit 2ddc278

Please sign in to comment.