Remove pinning & use CPU whls directly

Co-authored-by: Pavol Rusnak <pavol@rusnak.io>
This commit is contained in:
bsilvereagle 2023-04-02 14:30:57 -07:00 committed by GitHub
parent 6a498f0d79
commit 9fe1524d02
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -7,7 +7,7 @@ RUN apt-get update && \
RUN pip install --upgrade pip setuptools wheel \
&& pip install numpy requests sentencepiece tqdm \
&& pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
&& pip install torch --index-url https://download.pytorch.org/whl/cpu
WORKDIR /app