Merge remote-tracking branch 'refs/remotes/origin/dev' into dev

This commit is contained in:
oobabooga 2023-11-29 15:13:17 -08:00
commit fa89d305e3

View File

@ -45,7 +45,7 @@ RUN --mount=type=cache,target=/root/.cache/pip,rw \
. /app/venv/bin/activate && \
pip3 install --upgrade pip setuptools wheel ninja && \
pip3 install torch xformers --index-url https://download.pytorch.org/whl/cu121 && \
pip3 install torchvision torchaudio sentence_transformers
pip3 install torchvision torchaudio sentence-transformers
# Copy and install GPTQ-for-LLaMa
COPY --from=builder /build /app/repositories/GPTQ-for-LLaMa