Skip to content

Commit 7db41f8

Browse files
committed
remove cpu_offload unit test when enable lora due to #1817
Signed-off-by: ruit <ruit@nvidia.com>
1 parent cd882ec commit 7db41f8

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

‎tests/unit/models/generation/test_vllm_generation.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -900,9 +900,7 @@ async def run_hf_train_process(
900900
(False, True, "fp8", False),
901901
# LoRA tests
902902
(False, False, "bfloat16", True),
903-
(False, True, "bfloat16", True),
904903
(True, False, "bfloat16", True),
905-
(True, True, "bfloat16", True),
906904
],
907905
)
908906
async def test_vllm_generation_with_hf_training_colocated(
@@ -968,9 +966,7 @@ async def test_vllm_generation_with_hf_training_colocated(
968966
(False, True, "fp8", False),
969967
# LoRA tests
970968
(False, False, "bfloat16", True),
971-
(False, True, "bfloat16", True),
972969
(True, False, "bfloat16", True),
973-
(True, True, "bfloat16", True),
974970
],
975971
)
976972
async def test_vllm_generation_with_hf_training_non_colocated(

0 commit comments

Comments
 (0)