Skip to content

Fix cuda index#42897

Merged
MekkCyber merged 2 commits intomainfrom
fix-cuda-index
Dec 17, 2025
Merged

Fix cuda index#42897
MekkCyber merged 2 commits intomainfrom
fix-cuda-index

Conversation

@SunMarc
Copy link
Copy Markdown
Member

@SunMarc SunMarc commented Dec 16, 2025

What does this PR do?

This PR fixes this issue #42851. As we moved to multi-threaded loading by default, if the users passes device_map="cuda", it will always load the model on cuda:0 as in threaded mode all the default devices reset to 0.

Copy link
Copy Markdown
Contributor

@MekkCyber MekkCyber left a comment

Choose a reason for hiding this comment

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

Makes sense! Thanks

@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

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.

Copy link
Copy Markdown
Collaborator

@ArthurZucker ArthurZucker left a comment

Choose a reason for hiding this comment

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

LGTM

@MekkCyber MekkCyber merged commit 05c7e4a into main Dec 17, 2025
26 checks passed
@MekkCyber MekkCyber deleted the fix-cuda-index branch December 17, 2025 14:47
SangbumChoi pushed a commit to SangbumChoi/transformers that referenced this pull request Jan 23, 2026
fix index

Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
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