Akash4911 commited on
Commit
ca05e49
·
1 Parent(s): 7f4fce1

Fix: Add CPU torchvision to prevent pyannote from downloading GPU torch and hitting disk limits

Browse files
Files changed (1) hide show
  1. 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