Skip to content

Commit b84ef4a

Browse files
committed
Bump transformers version
1 parent 7a929e8 commit b84ef4a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

setup.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@
2828

2929
INSTALL_REQUIRE = [
3030
"torch>=1.11",
31-
"transformers>=4.36.0,<4.41.0",
32-
"optimum~=1.19",
31+
"transformers>=4.36.0,<4.42.0",
32+
"optimum @ git+https://github.com/huggingface/optimum.git@bump-transformers",
3333
"datasets>=1.4.0",
3434
"sentencepiece",
3535
"scipy",

tests/openvino/test_modeling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1675,7 +1675,7 @@ def test_compare_output_attentions(self, model_arch):
16751675
preprocessor = AutoFeatureExtractor.from_pretrained(model_id)
16761676
inputs = preprocessor(images=image, return_tensors="pt")
16771677

1678-
transformers_model = AutoModelForImageClassification.from_pretrained(model_id)
1678+
transformers_model = AutoModelForImageClassification.from_pretrained(model_id, attn_implementation="eager")
16791679
transformers_model.eval()
16801680
with torch.no_grad():
16811681
transformers_outputs = transformers_model(**inputs, output_attentions=True)

0 commit comments

Comments
 (0)