Skip to content

Commit dd7f3a6

Browse files
roychanRoy Chan
authored andcommitted
Fix logic in setting self.fsdp when it is False (huggingface#41974)
Co-authored-by: Roy Chan <[email protected]>
1 parent 4d0c905 commit dd7f3a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/training_args.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2655,7 +2655,7 @@ def set_dataloader(
26552655
return self
26562656

26572657
def _process_fsdp_args(self):
2658-
if self.fsdp is None:
2658+
if not self.fsdp:
26592659
self.fsdp = []
26602660
elif self.fsdp is True:
26612661
self.fsdp = [FSDPOption.FULL_SHARD]

0 commit comments

Comments
 (0)