Skip to content

Commit 9043078

Browse files
committed
use optimum from branch
1 parent 9cbb3b0 commit 9043078

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

optimum/exporters/openvino/__main__.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -213,8 +213,7 @@ def main_export(
213213
subfolder=subfolder,
214214
revision=revision,
215215
cache_dir=cache_dir,
216-
token=token,
217-
library_name=library_name,
216+
token=token
218217
)
219218
if library_name == "sentence_transformers":
220219
logger.warning(

optimum/exporters/openvino/model_configs.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def init_model_configs():
146146
# for model registration in auto transformers classses
147147
if importlib.util.find_spec("janus") is not None:
148148
try:
149-
from janus.models import MultiModalityCausalLM, VLChatProcessor
149+
from janus.models import MultiModalityCausalLM # noqa: F401
150150
except ImportError:
151151
pass
152152

setup.py

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

2929
INSTALL_REQUIRE = [
3030
"torch>=1.11",
31-
"optimum~=1.24",
3231
"transformers>=4.36,<4.49",
32+
"optimum @ git+https://github.com/eaidova/optimum.git@ea/avoid_lib_guessing_in_standartize_args",
3333
"datasets>=1.4.0",
3434
"sentencepiece",
3535
"setuptools",

0 commit comments

Comments
 (0)