Skip to content

Commit

Permalink
update llamacpp and diffusers
Browse files Browse the repository at this point in the history
  • Loading branch information
Josh-XT committed Jul 11, 2024
1 parent a327da4 commit fd836a0
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion cuda-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ requests==2.32.0
tiktoken==0.6.0
python-dotenv==1.0.1
beautifulsoup4==4.12.3
faster-whisper==1.0.2
faster-whisper==1.0.3
pydub==0.25.1
ffmpeg-python==0.2.0
torch==2.3.1+cu121
Expand Down
2 changes: 1 addition & 1 deletion cuda.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ ENV HOST=0.0.0.0 \
LLAMA_CUBLAS=1
COPY cuda-requirements.txt .
RUN python3 -m pip install --upgrade pip cmake scikit-build setuptools wheel --no-cache-dir && \
CMAKE_ARGS="-DLLAMA_CUDA=on" FORCE_CMAKE=1 pip install llama-cpp-python==0.2.79 --no-cache-dir && \
CMAKE_ARGS="-DLLAMA_CUDA=on" FORCE_CMAKE=1 pip install llama-cpp-python==0.2.82 --no-cache-dir && \
pip install --no-cache-dir -r cuda-requirements.txt
RUN git clone https://github.com/Josh-XT/DeepSeek-VL deepseek && \
cd deepseek && \
Expand Down
3 changes: 2 additions & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -24,4 +24,5 @@ openai
pdfplumber
optimum
onnx
llama-cpp-python==0.2.79
diffusers[torch]
llama-cpp-python==0.2.82

0 comments on commit fd836a0

Please sign in to comment.