diff options
-rw-r--r-- | .devops/full.Dockerfile | 5 | ||||
-rw-r--r-- | flake.nix | 1 |
2 files changed, 3 insertions, 3 deletions
diff --git a/.devops/full.Dockerfile b/.devops/full.Dockerfile index a75bc97..491d676 100644 --- a/.devops/full.Dockerfile +++ b/.devops/full.Dockerfile @@ -5,9 +5,10 @@ FROM ubuntu:$UBUNTU_VERSION as build RUN apt-get update && \ apt-get install -y build-essential python3 python3-pip +COPY requirements.txt requirements.txt + RUN pip install --upgrade pip setuptools wheel \ - && pip install numpy requests sentencepiece tqdm \ - && pip install torch --index-url https://download.pytorch.org/whl/cpu + && pip install -r requirements.txt WORKDIR /app @@ -10,7 +10,6 @@ inherit system; }; llama-python = pkgs.python310.withPackages (ps: with ps; [ - torch numpy sentencepiece ]); |