transformers
Fix: Cast prefetch_bucket_size to integer for deepspeed >= 0.15
#33402
Merged

Fix: Cast prefetch_bucket_size to integer for deepspeed >= 0.15 #33402

kiddj
kiddj Fix: Cast prefetch bucket size to integer in zero_optimization
258877c5
muellerzr
muellerzr approved these changes on 2024-09-10
muellerzr muellerzr requested a review from LysandreJik LysandreJik 1 year ago
LysandreJik
LysandreJik approved these changes on 2024-09-11
LysandreJik LysandreJik merged ecf7024b into main 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone