diff --git a/tests/core/test_trainer_builder.py b/tests/core/test_trainer_builder.py index d16001120..42c66a608 100644 --- a/tests/core/test_trainer_builder.py +++ b/tests/core/test_trainer_builder.py @@ -58,6 +58,7 @@ def fixture_base_cfg(): "dataloader_num_workers": 1, "dataloader_pin_memory": True, "dataloader_prefetch_factor": 2, + "sequence_parallel_degree": 1, # Dtype "fp16": False, "bf16": False,