transformers
Move num_items_in_batch to correct device before accelerator.gather
#40773
Merged
Overview
Commits
1
Changes
View On GitHub

Move num_items_in_batch to correct device before accelerator.gather #40773

SunMarc merged 1 commit into huggingface:main from ssharpe42:fix_avg_across_device
ssharpe42
ssharpe42 add device
6b03e1c3
SunMarc
SunMarc approved these changes on 2025-09-10
HuggingFaceDocBuilderDev
SunMarc SunMarc merged 3378e7da into main 256 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