after #545 we do not need torch, tqdm and requests in the dependencies
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
inherit system;
};
llama-python = pkgs.python310.withPackages (ps: with ps; [
- torch
numpy
sentencepiece
]);