Skip to content

Commit cd3e6b6

Browse files
committed
move on optimum main
1 parent 415b8b2 commit cd3e6b6

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
INSTALL_REQUIRE = [
3030
"torch>=1.11",
3131
"transformers>=4.36,<4.49",
32-
"optimum @ git+https://github.com/eaidova/optimum.git@ea/avoid_lib_guessing_in_standartize_args",
32+
"optimum @ git+https://github.com/huggingface/optimum.git",
3333
"datasets>=1.4.0",
3434
"sentencepiece",
3535
"setuptools",

tests/openvino/test_modeling.py

-1
Original file line numberDiff line numberDiff line change
@@ -2114,7 +2114,6 @@ class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase):
21142114
if is_transformers_version(">=", "4.40.0"):
21152115
SUPPORTED_ARCHITECTURES += ["llava_next", "nanollava"]
21162116
if is_transformers_version(">=", "4.45.0"):
2117-
21182117
SUPPORTED_ARCHITECTURES += ["janus", "minicpmv", "internvl2", "phi3_v", "qwen2_vl", "maira2"]
21192118
TASK = "image-text-to-text"
21202119
REMOTE_CODE_MODELS = ["internvl2", "minicpmv", "nanollava", "phi3_v", "maira2", "janus"]

0 commit comments

Comments
 (0)