fix

2025-03-14 01:42:10 +00:00
parent cb3a9e99a3
commit a6ef6c7764
3 changed files with 5 additions and 4 deletions
--- a/tests/test_exact_deduplication.py
+++ b/tests/test_exact_deduplication.py
@@ -280,6 +280,7 @@ class TestDeduplicateNonRL(unittest.TestCase):
                "batch_size": 10,
                "micro_batch_size": 10,
                "num_epochs": 1,
+                "sequence_parallel_degree": 1,
            }
        )