added containers dir

This commit is contained in:
2026-04-10 20:48:24 -04:00
parent 4a10a80ba0
commit 259e952afc
6 changed files with 32 additions and 70 deletions

View File

@@ -12,7 +12,8 @@ from typing import Annotated
import typer
from python.prompt_bench.vllm_container import check_gpu_free, start_vllm, stop_vllm
from python.prompt_bench.containers.lib import check_gpu_free
from python.prompt_bench.containers.vllm import start_vllm, stop_vllm
from python.prompt_bench.downloader import is_model_present
from python.prompt_bench.models import BenchmarkConfig
from python.prompt_bench.vllm_client import VLLMClient