Skip to content

Commit e5f9b6a

Browse files
authored
[None][fix] Fix TRT-python multi LoRA TP=2 test arguments (#8059)
Signed-off-by: Amit Zuker <[email protected]>
1 parent 31a1a5f commit e5f9b6a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unittest/llmapi/test_llm_multi_gpu.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,6 +277,8 @@ def test_llama_7b_multi_lora_tp2():
277277
LLM,
278278
enable_lora=True,
279279
build_config=BuildConfig(lora_config=lora_config),
280+
lora_config=lora_config,
281+
tensor_parallel_size=2,
280282
fast_build=True,
281283
kv_cache_config=global_kv_cache_config)
282284

0 commit comments

Comments
 (0)