Skip to content

Commit

Permalink
update transformers version
Browse files Browse the repository at this point in the history
  • Loading branch information
JessicaXYWang committed Jan 14, 2025
1 parent 56e623d commit efa6aa0
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,12 @@
if _TRANSFORMERS_AVAILABLE:
import transformers

_TRANSFORMERS_EQUAL_4_32_1 = transformers.__version__ == "4.32.1"
if _TRANSFORMERS_EQUAL_4_32_1:
_TRANSFORMERS_EQUAL_4_48_0 = transformers.__version__ == "4.48.0"
if _TRANSFORMERS_EQUAL_4_48_0:
from transformers import AutoTokenizer
else:
raise RuntimeError(
"transformers should be == 4.32.1, found: {}".format(
"transformers should be == 4.48.0, found: {}".format(
transformers.__version__
)
)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,12 +13,12 @@
if _TRANSFORMERS_AVAILABLE:
import transformers

_TRANSFORMERS_EQUAL_4_32_1 = transformers.__version__ == "4.32.1"
if _TRANSFORMERS_EQUAL_4_32_1:
_TRANSFORMERS_EQUAL_4_48_0 = transformers.__version__ == "4.48.0"
if _TRANSFORMERS_EQUAL_4_48_0:
from transformers import AutoModelForSequenceClassification
else:
raise RuntimeError(
"transformers should be == 4.32.1, found: {}".format(
"transformers should be == 4.48.0, found: {}".format(
transformers.__version__
)
)
Expand Down

0 comments on commit efa6aa0

Please sign in to comment.