aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPavol Rusnak <pavol@rusnak.io>2023-04-14 15:37:11 +0200
committerGitHub <noreply@github.com>2023-04-14 15:37:11 +0200
commita32f7acc9f54dba1c728cb1e596bd00bf3b4eb5f (patch)
tree1d486dc895e2b255f9b363a15a8ca4dd4b83c960
parent43ffdefb7424f79a3d510c199e2ea86684b4f824 (diff)
py : cleanup dependencies (#962)
after #545 we do not need torch, tqdm and requests in the dependencies
-rw-r--r--.devops/full.Dockerfile5
-rw-r--r--flake.nix1
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
diff --git a/flake.nix b/flake.nix
index 91d2edd..5363052 100644
--- a/flake.nix
+++ b/flake.nix
@@ -10,7 +10,6 @@
inherit system;
};
llama-python = pkgs.python310.withPackages (ps: with ps; [
- torch
numpy
sentencepiece
]);