-
Notifications
You must be signed in to change notification settings - Fork 31.8k
Attempt to fix Flax CI error(s) #8829
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
Conversation
Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
…weight from the hub. Signed-off-by: Morgan Funtowicz <[email protected]>
…cumenting. Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
Signed-off-by: Morgan Funtowicz <[email protected]>
patrickvonplaten
left a comment
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.
I like it a lot! makes the tests easier to read and seems to fix CI - thanks a lot!
LysandreJik
left a comment
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.
Cool! Thanks for looking into it.
Signed-off-by: Morgan Funtowicz <[email protected]>
sgugger
left a comment
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.
Thanks for fixing!
* Slightly increase tolerance between pytorch and flax output Signed-off-by: Morgan Funtowicz <[email protected]> * test_multiple_sentences doesn't require torch Signed-off-by: Morgan Funtowicz <[email protected]> * Simplify parameterization on "jit" to use boolean rather than str Signed-off-by: Morgan Funtowicz <[email protected]> * Use `require_torch` on `test_multiple_sentences` because we pull the weight from the hub. Signed-off-by: Morgan Funtowicz <[email protected]> * Rename "jit" parameter to "use_jit" for (hopefully) making it self-documenting. Signed-off-by: Morgan Funtowicz <[email protected]> * Remove pytest.mark.parametrize which seems to fail in some circumstances Signed-off-by: Morgan Funtowicz <[email protected]> * Fix unused imports. Signed-off-by: Morgan Funtowicz <[email protected]> * Fix style. Signed-off-by: Morgan Funtowicz <[email protected]> * Give default parameters values for traced model. Signed-off-by: Morgan Funtowicz <[email protected]> * Review comment: Change sentences to sequences Signed-off-by: Morgan Funtowicz <[email protected]> * Apply suggestions from code review Co-authored-by: Sylvain Gugger <[email protected]>
jitparametrization when runningtest_multiple_sentencesbecause it leads to instabilities