File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change 32
32
try :
33
33
from transformers import __version__ as _transformers_version
34
34
35
- if not (_Version ("4.48" ) <= _Version (_transformers_version ) < _Version ("5.0 " )):
35
+ if not (_Version ("4.48" ) <= _Version (_transformers_version ) < _Version ("4.57 " )):
36
36
_warnings .warn (
37
37
f"transformers version { _transformers_version } is incompatible with nvidia-modelopt and may cause issues. "
38
38
"Please install recommended version with `pip install nvidia-modelopt[hf]` if working with HF models." ,
Original file line number Diff line number Diff line change 62
62
"diffusers>=0.32.2" ,
63
63
"huggingface_hub>=0.24.0" ,
64
64
"peft>=0.17.0" ,
65
- "transformers>=4.48,<5.0 " , # Should match modelopt/torch/__init__.py and tox.ini
65
+ "transformers>=4.48,<4.57 " , # Should match modelopt/torch/__init__.py and tox.ini
66
66
"deepspeed>=0.9.6 ; platform_system != 'Darwin' and platform_system != 'Windows'" ,
67
67
],
68
68
# linter tools
Original file line number Diff line number Diff line change 27
27
28
28
# Should match setup.py
29
29
tf_min: transformers~=4.48.0
30
+ tf_latest: transformers~=4.56.0
30
31
commands =
31
32
python -m pytest tests/unit {env:COV_ARGS:}
32
33
You can’t perform that action at this time.
0 commit comments