File tree Expand file tree Collapse file tree 4 files changed +4
-3
lines changed Expand file tree Collapse file tree 4 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 11pyarrow
2- transformers < 5.0
2+ transformers < 4.58
33trl >= 0.23.0
Original file line number Diff line number Diff line change 3232try :
3333 from transformers import __version__ as _transformers_version
3434
35- if not (_Version ("4.48" ) <= _Version (_transformers_version ) < _Version ("5.0 " )):
35+ if not (_Version ("4.48" ) <= _Version (_transformers_version ) < _Version ("4.58 " )):
3636 _warnings .warn (
3737 f"transformers version { _transformers_version } is not tested with nvidia-modelopt and may cause issues. "
3838 "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 6161 "diffusers>=0.32.2" ,
6262 "huggingface_hub>=0.24.0" ,
6363 "peft>=0.17.0" ,
64- "transformers>=4.48,<5.0 " , # Should match modelopt/torch/__init__.py and tox.ini
64+ "transformers>=4.48,<4.58 " , # Should match modelopt/torch/__init__.py and tox.ini
6565 "deepspeed>=0.9.6 ; platform_system != 'Darwin' and platform_system != 'Windows'" ,
6666 ],
6767 # linter tools
Original file line number Diff line number Diff line change 2727
2828 # Should match setup.py
2929 tf_min: transformers~=4.48.0
30+ tf_latest: transformers~=4.57.0
3031commands =
3132 python -m pytest tests/unit {env:COV_ARGS:}
3233
You can’t perform that action at this time.
0 commit comments