Skip to content

[docs] Model API #3562

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 15 commits into from
Jun 30, 2023
Merged

[docs] Model API #3562

merged 15 commits into from
Jun 30, 2023

Conversation

stevhliu
Copy link
Member

@stevhliu stevhliu commented May 25, 2023

Creates a separate section for the models so that each model can have its own page and makes improvements to existing docstrings.

To do:

  • ModelMixin and FlaxModelMixin
  • UNet
  • VQModel
  • AutoEncoderKL and FlaxAutoEncoderKL
  • Transformer2D
  • TransformerTemporal
  • PriorTransformer
  • ControlNet and FlaxControlNet

@HuggingFaceDocBuilderDev
Copy link

HuggingFaceDocBuilderDev commented May 25, 2023

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

@stevhliu stevhliu force-pushed the api-maintenance branch 2 times, most recently from c543f90 to ddf5469 Compare May 30, 2023 22:55
@stevhliu
Copy link
Member Author

First draft of the model APIs is done (I'll do a second pass for typos and such before merging)! Some questions I had while working on this:

  • The source of the replace function in FlaxControlNetOutput leads to a 404
  • Some parameters refer to classes that aren't currently documented; for example, the ImagePositionalEmbeddings in the sample_size parameter isn't documented. Should we expose the Embeddings classes here?
  • Some models like ResNet, DualTransformer2D, and T5Film aren't included either. Is this on purpose or should they also be documented?

Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

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

Looking great!

Did a pass around the mdx files.

@stevhliu stevhliu marked this pull request as ready for review June 12, 2023 17:38
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.

Super nice! Great effort on cleaning all these docs!

@stevhliu stevhliu merged commit 174dcd6 into huggingface:main Jun 30, 2023
@stevhliu stevhliu deleted the api-maintenance branch June 30, 2023 00:24
yoonseokjin pushed a commit to yoonseokjin/diffusers that referenced this pull request Dec 25, 2023
* add modelmixin and unets

* remove old model page

* minor fixes

* fix unet2dcondition

* add vqmodel and autoencoderkl

* add rest of models

* fix autoencoderkl path

* fix toctree

* fix toctree again

* apply feedback

* apply feedback

* fix copies

* fix controlnet copy

* fix copies
AmericanPresidentJimmyCarter pushed a commit to AmericanPresidentJimmyCarter/diffusers that referenced this pull request Apr 26, 2024
* add modelmixin and unets

* remove old model page

* minor fixes

* fix unet2dcondition

* add vqmodel and autoencoderkl

* add rest of models

* fix autoencoderkl path

* fix toctree

* fix toctree again

* apply feedback

* apply feedback

* fix copies

* fix controlnet copy

* fix copies
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