Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

[docs] Model sharding #9521

Merged
merged 2 commits into from
Sep 25, 2024
Merged

[docs] Model sharding #9521

merged 2 commits into from
Sep 25, 2024

Conversation

stevhliu
Copy link
Member

Adds doc about model sharding (selectively loading/unloading components, device_map) based on @sayakpaul's write up!

@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 September 24, 2024 19:35
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.

Very nice, thanks Steven! I left a few minor comments.

@stevhliu stevhliu merged commit d9c9691 into huggingface:main Sep 25, 2024
1 check passed
@stevhliu stevhliu deleted the sharded-pipeline branch September 25, 2024 16:33
leisuzz pushed a commit to leisuzz/diffusers that referenced this pull request Oct 11, 2024
sayakpaul pushed a commit that referenced this pull request Dec 23, 2024
* flux shard

* feedback
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants