fixed prompt_bench.py

2026-04-28 22:48:54 -04:00
parent 923d128f00
commit 97f0083ca2
1 changed files with 2 additions and 2 deletions
@@ -11,8 +11,8 @@ from typing import Annotated

 import typer

-from pipelines.tools.containers.lib import check_gpu_free
-from pipelines.tools.containers.vllm import start_vllm, stop_vllm
+from pipelines.containers.lib import check_gpu_free
+from pipelines.containers.vllm import start_vllm, stop_vllm
 from pipelines.tools.downloader import is_model_present
 from pipelines.tools.models import BenchmarkConfig
 from pipelines.tools.vllm_client import VLLMClient