transformers
Add new dim to `num_items_in_batch` if necessary
#36967
Merged
Overview
Commits
8
Changes
View On GitHub

Add new dim to `num_items_in_batch` if necessary #36967

ydshieh merged 8 commits into huggingface:main from regisss:fix_multi_gpu_num_items
regisss
regisss Add new dim to `num_items_in_batch` if necessary
a2c0f0ac
github-actions
github-actions github-actions marked this pull request as draft 1 year ago
HuggingFaceDocBuilderDev
regisss regisss marked this pull request as ready for review 1 year ago
regisss regisss requested a review from ArthurZucker ArthurZucker 1 year ago
Rocketknight1
IlyasMoutawwakil
BenjaminBossan
regisss Merge branch 'main' into fix_multi_gpu_num_items
3d5d0dd7
regisss Unsqueeze only in the DP case
a26ed23e
regisss
Mekadrom
IlyasMoutawwakil Merge branch 'main' into fix_multi_gpu_num_items
ec19fefa
IlyasMoutawwakil
IlyasMoutawwakil approved these changes on 2025-04-01
BenjaminBossan
IlyasMoutawwakil Merge branch 'main' into fix_multi_gpu_num_items
912afebd
ydshieh ydshieh requested a review from muellerzr muellerzr 359 days ago
IlyasMoutawwakil Merge branch 'main' into fix_multi_gpu_num_items
fbcf5bde
SunMarc
SunMarc approved these changes on 2025-04-02
SunMarc Merge branch 'main' into fix_multi_gpu_num_items
91d6cb74
ydshieh Merge branch 'main' into fix_multi_gpu_num_items
c161aecc
ydshieh ydshieh merged 12048990 into main 358 days ago
regisss regisss deleted the fix_multi_gpu_num_items branch 358 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum