py : cleanup dependencies (#962)

after #545 we do not need torch, tqdm and requests in the dependencies
This commit is contained in:
Pavol Rusnak 2023-04-14 15:37:11 +02:00 committed by GitHub
parent 43ffdefb74
commit a32f7acc9f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions

View File

@ -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 -r requirements.txt
&& pip install torch --index-url https://download.pytorch.org/whl/cpu
WORKDIR /app WORKDIR /app

View File

@ -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
]); ]);