Skip to content

[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

Merged
merged 5 commits into from
Feb 8, 2023
Merged

Conversation

patrickvonplaten
Copy link
Contributor

@patrickvonplaten patrickvonplaten commented Feb 8, 2023

This PR accelerates the tests for:

  • textual inversion
  • unconditional
    and adds new tests for
  • dreambooth
  • text_to_image

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.

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented Feb 8, 2023

The documentation is not available anymore as the PR was closed or merged.

Copy link
Member

@pcuenca pcuenca left a 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!

Comment on lines +70 to +75
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.",
)
Copy link
Member

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
Copy link
Member

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.

Copy link
Contributor

@patil-suraj patil-suraj left a 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!

@patrickvonplaten patrickvonplaten merged commit 1ed6b77 into main Feb 8, 2023
@patrickvonplaten patrickvonplaten deleted the add_more_example_tests branch February 8, 2023 14:59
yiyixuxu pushed a commit to evinpinar/diffusers-attend-and-excite-pipeline that referenced this pull request Feb 16, 2023
* [Examples] Test all examples on CPU

* add

* correct

* Apply suggestions from code review
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
* [Examples] Test all examples on CPU

* add

* correct

* Apply suggestions from code review
AmericanPresidentJimmyCarter pushed a commit to AmericanPresidentJimmyCarter/diffusers that referenced this pull request Apr 26, 2024
* [Examples] Test all examples on CPU

* add

* correct

* Apply suggestions from code review
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants