results

mosaicml · May 8, 2024 · d38c254 · d38c254
1 parent 1dd4245
commit d38c254
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/utils/test_autolog_hparams.py b/tests/utils/test_autolog_hparams.py
@@ -178,6 +178,7 @@ def test_extract_hparams_trainer():
         'deepspeed_config': None,
         'fsdp_config': None,
         'fsdp_auto_wrap': True,
+        'tp_config': None,
 
         # System/Numerics
         'device': 'DeviceCPU',