-
Notifications
You must be signed in to change notification settings - Fork 6.7k
[Examples] Test all examples on CPU #2289
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
|
The documentation is not available anymore as the PR was closed or merged. |
pcuenca
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.
Great work, thanks a lot!
| parser.add_argument( | ||
| "--model_config_name_or_path", | ||
| type=str, | ||
| default=None, | ||
| help="The config of the UNet model to train, leave as None to use standard DDPM configuration.", | ||
| ) |
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.
Nice!
| from diffusers.optimization import get_scheduler | ||
| from diffusers.training_utils import EMAModel | ||
| from diffusers.utils import check_min_version | ||
| from diffusers.utils import check_min_version, is_tensorboard_available |
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.
tensorboard is in the requirements for the text_to_image and textual_inversion examples, but I actually prefer it this way.
patil-suraj
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.
Looks good, thanks a lot!
* [Examples] Test all examples on CPU * add * correct * Apply suggestions from code review
* [Examples] Test all examples on CPU * add * correct * Apply suggestions from code review
* [Examples] Test all examples on CPU * add * correct * Apply suggestions from code review
This PR accelerates the tests for:
and adds new tests for
and makes sure those tests are also run on PRs on CPU. In my local test they should take around ~1min which is ok IMO.