Skip to content

Commit b649648

Browse files
committed
fix: Use enforce_eager=True for test
Signed-off-by: Sanger Steel <sangersteel@gmail.com>
1 parent 75d5681 commit b649648

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/lora/test_llama_tp.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -226,6 +226,7 @@ def test_serialize_and_deserialize_lora(tmp_path, sql_lora_files,
226226
loaded_vllm_model = LLM(model=model_ref,
227227
load_format="tensorizer",
228228
enable_lora=True,
229+
enforce_eager=True,
229230
model_loader_extra_config=tensorizer_config,
230231
max_num_seqs=13,
231232
max_loras=2,

0 commit comments

Comments
 (0)