From 37fb68ac7e9825665414d6f866aa4992e958c66d Mon Sep 17 00:00:00 2001 From: Richie Cahill Date: Tue, 28 Apr 2026 22:48:54 -0400 Subject: [PATCH] fixed prompt_bench.py --- pipelines/tools/prompt_bench.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pipelines/tools/prompt_bench.py b/pipelines/tools/prompt_bench.py index 60391c5..c126fc5 100644 --- a/pipelines/tools/prompt_bench.py +++ b/pipelines/tools/prompt_bench.py @@ -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