-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
Add support for model.compile
in PyTorch 2.0
#3246
Conversation
tgaddair
commented
Mar 14, 2023
ludwig/schema/trainer.py
Outdated
raise ConfigValidationError( | ||
"ECD model training requires PyTorch 2.0.0 or higher. Please upgrade PyTorch and try again." | ||
) |
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 would change this to compile requires PyTorch 2.0.0 or higher. Please upgrade PyTorch and try again
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.
That's what I get for letting Copilot write the error messages.
…nto torch-compile