frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ci] pin numpy to < 2 on win (#2009)

Browse Source
This commit is contained in:
Kashif Rasul
2024-09-03 13:03:38 +02:00
committed by GitHub
parent 728a9a3b5f
commit d60a1f50fe
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -64,7 +64,8 @@ __version__ = "0.11.0.dev0" # expected format is one of x.y.z.dev0, or x.y.z.rc
REQUIRED_PKGS = [
"torch>=1.4.0",
"transformers>=4.31.0",
"numpy>=1.18.2",
"numpy>=1.18.2;platform_system!='Windows'",
"numpy<2;platform_system=='Windows'",
"accelerate",
"datasets",
"tyro>=0.5.11",