Jelajahi Sumber

py : cleanup dependencies (#962)

after #545 we do not need torch, tqdm and requests in the dependencies
Pavol Rusnak 2 tahun lalu
induk
melakukan
a32f7acc9f
2 mengubah file dengan 3 tambahan dan 3 penghapusan
  1. 3 2
      .devops/full.Dockerfile
  2. 0 1
      flake.nix

+ 3 - 2
.devops/full.Dockerfile

@@ -5,9 +5,10 @@ FROM ubuntu:$UBUNTU_VERSION as build
 RUN apt-get update && \
 RUN apt-get update && \
     apt-get install -y build-essential python3 python3-pip
     apt-get install -y build-essential python3 python3-pip
 
 
+COPY requirements.txt requirements.txt
+
 RUN pip install --upgrade pip setuptools wheel \
 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
 WORKDIR /app
 
 

+ 0 - 1
flake.nix

@@ -10,7 +10,6 @@
           inherit system;
           inherit system;
         };
         };
         llama-python = pkgs.python310.withPackages (ps: with ps; [
         llama-python = pkgs.python310.withPackages (ps: with ps; [
-          torch
           numpy
           numpy
           sentencepiece
           sentencepiece
         ]);
         ]);