-
Notifications
You must be signed in to change notification settings - Fork 6.1k
[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. |
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!
@@ -22,7 +22,7 @@ | |||
from diffusers import DDPMPipeline, DDPMScheduler, UNet2DModel | |||
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.
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.