Spaces:
Sleeping
Sleeping
Fix: Add CPU torchvision to prevent pyannote from downloading GPU torch and hitting disk limits
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -23,7 +23,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
|
|
| 23 |
pip install --no-cache-dir fastapi "uvicorn[standard]" python-multipart python-dotenv pydantic "pydantic-settings" motor "passlib[bcrypt]" PyJWT google-generativeai
|
| 24 |
|
| 25 |
# Step 2: PyTorch CPU-only (large but isolated for caching)
|
| 26 |
-
RUN pip install --no-cache-dir torch torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 27 |
|
| 28 |
# Step 3: ML/AI packages
|
| 29 |
RUN pip install --no-cache-dir transformers accelerate "sentence-transformers" scikit-learn numpy scipy
|
|
@@ -32,7 +32,7 @@ RUN pip install --no-cache-dir transformers accelerate "sentence-transformers" s
|
|
| 32 |
RUN pip install --no-cache-dir Pillow piexif opencv-python-headless "soundfile>=0.12.0" "librosa>=0.10.0" "resampy>=0.4.2"
|
| 33 |
|
| 34 |
# Step 5: Heavy optional packages
|
| 35 |
-
RUN pip install --no-cache-dir "pyannote.audio>=3.1.0" "protobuf~=4.25.3" python-magic
|
| 36 |
|
| 37 |
COPY . .
|
| 38 |
|
|
|
|
| 23 |
pip install --no-cache-dir fastapi "uvicorn[standard]" python-multipart python-dotenv pydantic "pydantic-settings" motor "passlib[bcrypt]" PyJWT google-generativeai
|
| 24 |
|
| 25 |
# Step 2: PyTorch CPU-only (large but isolated for caching)
|
| 26 |
+
RUN pip install --no-cache-dir torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
|
| 27 |
|
| 28 |
# Step 3: ML/AI packages
|
| 29 |
RUN pip install --no-cache-dir transformers accelerate "sentence-transformers" scikit-learn numpy scipy
|
|
|
|
| 32 |
RUN pip install --no-cache-dir Pillow piexif opencv-python-headless "soundfile>=0.12.0" "librosa>=0.10.0" "resampy>=0.4.2"
|
| 33 |
|
| 34 |
# Step 5: Heavy optional packages
|
| 35 |
+
RUN pip install --no-cache-dir "pyannote.audio>=3.1.0" "protobuf~=4.25.3" python-magic c2pa-python
|
| 36 |
|
| 37 |
COPY . .
|
| 38 |
|