fix ci

li126com · Jun 19, 2024 · 66498c8 · 66498c8
1 parent 2de146b
commit 66498c8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_data/test_batch_sampler.py b/tests/test_data/test_batch_sampler.py
@@ -187,6 +187,7 @@ def test_warmup(use_flash_atten_case, group_case, micro_bsz_case):
     config.data.gradient_accumulation = config.data.micro_num
     config.data.rampup_batch_size = group_case[1]
     config.data.packed_length = micro_bsz_case * config.data.seq_len
+    config.data.use_shm = False
     should_sccuess = group_case[2]
     answer = group_case[3]