transformers
Default `synced_gpus` to `True` when using `FullyShardedDataParallel`
#33483
Merged

Default `synced_gpus` to `True` when using `FullyShardedDataParallel` #33483

ringohoffman
Default synced_gpus to True when using FullyShardedDataParallel
7179e614
Remove test file
823a44b3
ruff formatting
80af5a04
ruff format
4cfee750
ringohoffman
ringohoffman ringohoffman changed the title Default synced_gpus to True when using FullyShardedDataParallel Default `synced_gpus` to `True` when using `FullyShardedDataParallel` 1 year ago
ringohoffman
Merge branch 'main' into fsdp-generate-synced_gpus-default-to-true
24fecab4
ringohoffman
LysandreJik
ArthurZucker
ArthurZucker approved these changes on 2024-10-03
Update copyright year
b06d7a02
Merge branch 'main' into fsdp-generate-synced_gpus-default-to-true
ed1bb713
Add test for FSDP-wrapped model generation
5811565d
Ruff format
7183b247
ringohoffman
Move argparse import
aadaddc2
Remove barrier
c0468c40
SunMarc
SunMarc approved these changes on 2024-10-04
Move import into function to decrease load time
22680991
SunMarc SunMarc requested a review from muellerzr muellerzr 1 year ago
muellerzr
muellerzr approved these changes on 2024-10-07
Add test for accelerate and Trainer
fb1805de
Refactor imports
c0f2b5fb
Ruff format
7beca831
Use nullcontext
d81ca75b
Merge branch 'main' into fsdp-generate-synced_gpus-default-to-true
84cafb72
muellerzr
muellerzr approved these changes on 2024-10-10
muellerzr muellerzr merged 70b07d97 into main 1 year ago
ringohoffman ringohoffman deleted the fsdp-generate-synced_gpus-default-to-true branch 1 year ago
YeLuoSuiYou
ringohoffman
YeLuoSuiYou
YeLuoSuiYou

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone