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 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.57 " )):
3636 _warnings .warn (
3737 f"transformers version { _transformers_version } is incompatible 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 6262 "diffusers>=0.32.2" ,
6363 "huggingface_hub>=0.24.0" ,
6464 "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
6666 "deepspeed>=0.9.6 ; platform_system != 'Darwin' and platform_system != 'Windows'" ,
6767 ],
6868 # 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.56.0
3031commands =
3132 python -m pytest tests/unit {env:COV_ARGS:}
3233
You can’t perform that action at this time.
0 commit comments