Skip to content

Commit 75d5681

Browse files
committed
fix: Get correct absolute path to example script
Signed-off-by: Sanger Steel <sangersteel@gmail.com>
1 parent 8b1bb8e commit 75d5681

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/lora/test_llama_tp.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from vllm.lora.request import LoRARequest
1212
from vllm.model_executor.model_loader.tensorizer import TensorizerConfig
1313

14-
from ..utils import create_new_process_for_each_test, multi_gpu_test
14+
from ..utils import VLLM_PATH, create_new_process_for_each_test, multi_gpu_test
1515

1616
MODEL_PATH = "meta-llama/Llama-2-7b-hf"
1717

@@ -202,8 +202,9 @@ def test_serialize_and_deserialize_lora(tmp_path, sql_lora_files,
202202
suffix = "test"
203203
try:
204204
result = subprocess.run([
205-
sys.executable, "examples/other/tensorize_vllm_model.py",
206-
"--model", MODEL_PATH, "--lora-path", lora_path, "serialize",
205+
sys.executable,
206+
f"{VLLM_PATH}/examples/other/tensorize_vllm_model.py", "--model",
207+
MODEL_PATH, "--lora-path", lora_path, "serialize",
207208
"--serialized-directory",
208209
str(tmp_path), "--suffix", suffix
209210
],

0 commit comments

Comments
 (0)