Skip to content
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

[docs] Quantization tip #10249

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

[docs] Quantization tip #10249

wants to merge 4 commits into from

Conversation

stevhliu
Copy link
Member

@stevhliu stevhliu commented Dec 16, 2024

Follows up on discussion about adding a quantization section for big models. Instead of adding it to the individual model doc (for example, MochiTransformer3DModel) and the pipeline doc, I think it's more discoverable/cleaner to add it only to the pipeline doc (for example, MochiPipeline).

Let me know if this works for you, and then I can add it to the other big models!

  • AuraFlow
  • CogVideoX
  • Flux
  • HunyuanVideo
  • Mochi
  • SD3

@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.

@stevhliu stevhliu requested a review from sayakpaul December 16, 2024 23:16
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.

I left two broad comments and your plan looks perfect to me!

@a-r-r-o-w WDYT?

docs/source/en/api/pipelines/mochi.md Outdated Show resolved Hide resolved
docs/source/en/api/pipelines/mochi.md Show resolved Hide resolved
@sayakpaul
Copy link
Member

sayakpaul commented Dec 20, 2024

I think the following classes are remaining:

We could either open this up to the community or tackle it ourselves in this PR.

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.

Thanks for the changes!

Two broad comments. Once they are addressed I think we should be good to ship this 🚀

docs/source/en/api/pipelines/aura_flow.md Outdated Show resolved Hide resolved
docs/source/en/api/pipelines/cogvideox.md Outdated Show resolved Hide resolved
@stevhliu
Copy link
Member Author

Thanks @sayakpaul! I included the latest batch of pipelines (Allegro, Latte, LTX, etc.) in this PR as well 🙂

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