Spaces:
Running
on
A100
Running
on
A100
fix ffmpeg issue
Browse files- Dockerfile +3 -3
- 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
|
| 14 |
-
#
|
| 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
|