Spaces:

GameScribes
/

Multipurpose-AI-Agent-Development

Sleeping

devve1 commited on Jul 10

Commit

17a5f1f

•

1 Parent(s): 148e5d5

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -29,7 +29,8 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH"
 COPY ./requirements.txt /data/requirements.txt
 RUN pip3 install --no-cache-dir --upgrade -r /data/requirements.txt
-RUN CMAKE_ARGS="-DLLAMA_CUDA=on" pip3 install --no-cache-dir llama-cpp-python
 COPY pip.conf $VIRTUAL_ENV/pip.conf

 COPY ./requirements.txt /data/requirements.txt
 RUN pip3 install --no-cache-dir --upgrade -r /data/requirements.txt
+RUN pip3 install --no-cache-dir llama-cpp-python \
+    --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/<cuda-version>
 COPY pip.conf $VIRTUAL_ENV/pip.conf