|
@@ -6,7 +6,8 @@ RUN apt-get update && \
|
|
|
apt-get install -y build-essential python3 python3-pip
|
|
apt-get install -y build-essential python3 python3-pip
|
|
|
|
|
|
|
|
RUN pip install --upgrade pip setuptools wheel \
|
|
RUN pip install --upgrade pip setuptools wheel \
|
|
|
- && pip install numpy requests sentencepiece torch tqdm
|
|
|
|
|
|
|
+ && pip install numpy requests sentencepiece tqdm \
|
|
|
|
|
+ && pip install torch --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
|
|
|
WORKDIR /app
|
|
WORKDIR /app
|
|
|
|
|
|