fix: tf

open-webui · May 31, 2024 · c68701d · c68701d
1 parent 2bb2cba
commit c68701d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -25,12 +25,9 @@ WORKDIR /app
 # Install Python dependencies
 RUN pip3 install uv && \
     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 tensorflow[and-cuda] --no-cache-dir && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     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 && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     fi

diff --git a/requirements.txt b/requirements.txt
@@ -34,7 +34,6 @@ langchain-community
 langfuse
 
 # ML libraries
-tensorflow
 torch
 xgboost
-Original file line number
+Diff line change
@@ Expand Up / @@ -34,7 +34,6 @@ langchain-community @@
     langfuse
     # ML libraries
-    tensorflow
     torch
     xgboost
@@ Expand Down @@