alihan/llava-docker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Install correct version of Torch

Browse Source
This commit is contained in:
Ashley Kleynhans
2023-10-20 10:09:28 +02:00
parent bfc8345434
commit 1d71c18a84
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@@ -63,6 +63,12 @@ FROM base as setup
# Create and use the Python venv # Create and use the Python venv
RUN python3 -m venv /venv RUN python3 -m venv /venv
# Install Torch
RUN source /venv/bin/activate && \
pip3 install --no-cache-dir torch==2.0.1 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 && \
pip3 install --no-cache-dir xformers==0.0.22 && \
deactivate
# Clone the git repo of LLaVA and set version # Clone the git repo of LLaVA and set version
WORKDIR / WORKDIR /
RUN git clone https://github.com/ashleykleynhans/LLaVA.git RUN git clone https://github.com/ashleykleynhans/LLaVA.git

2
README.md
View File

@@ -7,7 +7,7 @@
* Python 3.10.12 * Python 3.10.12
* [LLaVA]( * [LLaVA](
https://github.com/haotian-liu/llava) v1.1.1 https://github.com/haotian-liu/llava) v1.1.1
* Torch 2.1.0 * Torch 2.0.1
* BakLLaVA-1 model * BakLLaVA-1 model
## Available on RunPod ## Available on RunPod