devve1 commited on
Commit
3da0747
1 Parent(s): 17a5f1f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -30,7 +30,7 @@ COPY ./requirements.txt /data/requirements.txt
30
  RUN pip3 install --no-cache-dir --upgrade -r /data/requirements.txt
31
 
32
  RUN pip3 install --no-cache-dir llama-cpp-python \
33
- --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/<cuda-version>
34
 
35
  COPY pip.conf $VIRTUAL_ENV/pip.conf
36
 
 
30
  RUN pip3 install --no-cache-dir --upgrade -r /data/requirements.txt
31
 
32
  RUN pip3 install --no-cache-dir llama-cpp-python \
33
+ --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu122
34
 
35
  COPY pip.conf $VIRTUAL_ENV/pip.conf
36