Skip to content

Add UFOGenScheduler to Community Examples #6650

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
Jan 26, 2024

Conversation

dg845
Copy link
Contributor

@dg845 dg845 commented Jan 21, 2024

What does this PR do?

This PR adds a scheduler for the UFOGen model to the community examples. A separate UFOGen pipeline is not currently implemented because UFOGen is intended to be initialized from a pretrained diffusion model (Stable Diffusion 1.5 is used in the paper) and so UFOGenScheduler is designed to be compatible with existing pipelines such as StableDiffusionPipeline.

Based on #6133. The code in scheduling_ufogen.py is the same as in that PR, except with the diffusers imports changed from relative imports to absolute imports.

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@patrickvonplaten
@patil-suraj
@JunbongJang

Copy link
Contributor

@patrickvonplaten patrickvonplaten left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@dg845
Copy link
Contributor Author

dg845 commented Jan 24, 2024

Added an entry into the community README for scheduling_ufogen.py/UFOGenScheduler with a usage example (although the usage example doesn't quite work in the sense that the ufogen_model_id_or_path is just a dummy value rather than pointing to a physical trained UFOGen checkpoint, since I'm not aware of any publicly available such checkpoints).

@patrickvonplaten patrickvonplaten merged commit ea9dc3f into huggingface:main Jan 26, 2024
@dg845 dg845 deleted the ufogen-community-pipeline branch January 30, 2024 02:50
AmericanPresidentJimmyCarter pushed a commit to AmericanPresidentJimmyCarter/diffusers that referenced this pull request Apr 26, 2024
* Add UFOGenScheduler with diffusers imports changed from relative to absolute.

* make style

* Add community README entry for UFOGenScheduler.
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.

3 participants