-
Notifications
You must be signed in to change notification settings - Fork 41
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
requirements: Adjust numpy dependency range #68
Conversation
requirements.txt
Outdated
@@ -7,7 +7,7 @@ torch>=2.3.0a0 | |||
transformers>=4.41.2 | |||
datasets>=2.15.0 | |||
numba | |||
numpy==1.26.4 | |||
numpy>=1.26.4,<2.0.0 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please change this line to:
numpy>=1.26.4,<2.0.0 | |
numpy<2 |
I believe the 1.26.4
was arbitrarily pinned, it just needs to be under 2.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
got it. I'm actually going to copy over what the CLI repo has, because it had some more specific reasons on versions
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just need to change how we're pinning the version but otherwise LGTM
According to comments on instructlab#61, numpy 2 doesn't work with this library right now. Instead of pinning to a specific 1.x version, allow a range up to, but not including 2.0.0. The specifics on the lower end come from what's used in `instructlab/instructlab`. Signed-off-by: Russell Bryant <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm
According to comments on #61, numpy 2 doesn't work with this library
right now. Instead of pinning to a specific 1.x version, allow a range
from a known working version up to, but not including 2.0.0.
Signed-off-by: Russell Bryant [email protected]