accelerate
add support for custom function for reducing the batch size
#3071

Merged

add support for custom function for reducing the batch size #3071

SunMarc merged 3 commits into huggingface:main from winglian:auto_batch_size_callback

add support for custom function for reducing the batch size

dd1538e5

fix scoping

24d20b24

muellerzr approved these changes on 2024-09-03

muellerzr requested a review from

SunMarc 1 year ago

SunMarc approved these changes on 2024-09-04

muellerzr added wip

Apply style fixes

1e781af7

SunMarc closed this 362 days ago

SunMarc reopened this 362 days ago

SunMarc merged 650b6659 into main 362 days ago

Reviewers

SunMarc

muellerzr

Assignees

No one assigned

Labels

wip

Milestone

No milestone