ChuxiJ commited on
Commit
6fafd1b
·
1 Parent(s): 5a6536d

fix ffmpeg issue

Browse files
Files changed (2) hide show
  1. Dockerfile +3 -3
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -10,11 +10,11 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
10
 
11
  # Install system dependencies
12
  # build-essential is required for triton to compile CUDA kernels
13
- # ffmpeg is installed WITHOUT --no-install-recommends to ensure all shared libraries
14
- # (libavcodec, libavformat, libavutil, libswresample) are installed for torchaudio
15
  RUN apt-get update && \
16
  apt-get install -y --no-install-recommends git libsndfile1 build-essential && \
17
- apt-get install -y ffmpeg && \
18
  rm -rf /var/lib/apt/lists/*
19
 
20
  # Set up a new user named "user" with user ID 1000 (HuggingFace Space requirement)
 
10
 
11
  # Install system dependencies
12
  # build-essential is required for triton to compile CUDA kernels
13
+ # ffmpeg and libav* dev packages are required for torchaudio's ffmpeg backend
14
+ # Note: torchaudio's ffmpeg backend needs shared libraries, not just the ffmpeg binary
15
  RUN apt-get update && \
16
  apt-get install -y --no-install-recommends git libsndfile1 build-essential && \
17
+ apt-get install -y ffmpeg libavcodec-dev libavformat-dev libavutil-dev libswresample-dev && \
18
  rm -rf /var/lib/apt/lists/*
19
 
20
  # Set up a new user named "user" with user ID 1000 (HuggingFace Space requirement)
requirements.txt CHANGED
@@ -22,7 +22,7 @@ diskcache
22
  uvicorn[standard]>=0.27.0
23
  numba>=0.63.1
24
  vector-quantize-pytorch>=1.27.15
25
- torchcodec>=0.9.1
26
 
27
  # LoRA Training dependencies (optional)
28
  peft>=0.7.0
 
22
  uvicorn[standard]>=0.27.0
23
  numba>=0.63.1
24
  vector-quantize-pytorch>=1.27.15
25
+ # torchcodec>=0.9.1 # Disabled: causes CUDA dependency issues on HuggingFace Space
26
 
27
  # LoRA Training dependencies (optional)
28
  peft>=0.7.0