bumP

mosaicml · Aug 22, 2024 · f2e3496 · f2e3496
1 parent 416a1b7
commit f2e3496
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py
@@ -1322,7 +1322,7 @@ def __init__(
                 if isinstance(parallelism_config['fsdp'], FSDPConfig):
                     parallelism_config_args['fsdp'] = parallelism_config['fsdp']
                 else:
-                    parallelism_config_args['fsdp'] = FSDPConfig(**parallelism_config_args['fsdp'])
+                    parallelism_config_args['fsdp'] = FSDPConfig(**parallelism_config['fsdp'])
             if 'tp' in parallelism_config and parallelism_config['tp'] is not None:
                 if isinstance(parallelism_config['tp'], TPConfig):
                     parallelism_config_args['tp'] = parallelism_config['tp']