OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-12 16:40:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: tf

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-30 20:01:33 -07:00
parent 2bb2cba8c0
commit c68701da4b
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -25,12 +25,9 @@ WORKDIR /app
# Install Python dependencies # Install Python dependencies
RUN pip3 install uv && \ RUN pip3 install uv && \
if [ "$USE_CUDA" = "true" ]; then \ if [ "$USE_CUDA" = "true" ]; then \
pip3 install --no-binary=h5py h5py --no-cache-dir && \
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$USE_CUDA_DOCKER_VER --no-cache-dir && \ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$USE_CUDA_DOCKER_VER --no-cache-dir && \
pip3 install tensorflow[and-cuda] --no-cache-dir && \
uv pip install --system -r requirements.txt --no-cache-dir; \ uv pip install --system -r requirements.txt --no-cache-dir; \
else \ else \
pip3 install --no-binary=h5py h5py --no-cache-dir && \
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \
uv pip install --system -r requirements.txt --no-cache-dir; \ uv pip install --system -r requirements.txt --no-cache-dir; \
fi fi

1
requirements.txt
View File

@ -34,7 +34,6 @@ langchain-community
langfuse langfuse
# ML libraries # ML libraries
tensorflow
torch torch
xgboost xgboost