diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index 2c915bf9..00000000 --- a/requirements.txt +++ /dev/null @@ -1,14 +0,0 @@ -torch>=2.7.1 -accelerate>=0.26.0 -sentence-transformers>=2.2.0 -transformers>=4.54.0 -datasets>=2.0.0 -scikit-learn>=1.0.0 -numpy>=1.21.0 -pandas>=1.3.0 -requests>=2.25.0 -huggingface-hub>=0.10.0 -psutil>=7.0.0 -matplotlib>=3.10 -seaborn>=0.13 -openai>=1.100 diff --git a/tools/mock-vllm/Dockerfile b/tools/mock-vllm/Dockerfile index c11141d1..2af352bb 100644 --- a/tools/mock-vllm/Dockerfile +++ b/tools/mock-vllm/Dockerfile @@ -6,7 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ curl \ && rm -rf /var/lib/apt/lists/* -COPY requirements.txt ./ +COPY src/training/requirements.txt ./ RUN pip install --no-cache-dir -r requirements.txt COPY app.py ./