Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +5 -12
Dockerfile
CHANGED
@@ -1,12 +1,5 @@
|
|
1 |
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
|
2 |
|
3 |
-
# Accept HF token as build argument and set as environment variable
|
4 |
-
ARG HF_TOKEN
|
5 |
-
ENV HF_TOKEN=${HF_TOKEN}
|
6 |
-
|
7 |
-
# Debug print (you can remove this later)
|
8 |
-
RUN python -c "import os; print('✅ HF_TOKEN set' if os.getenv('HF_TOKEN') else '❌ HF_TOKEN missing')"
|
9 |
-
|
10 |
# Install OS-level dependencies
|
11 |
RUN apt-get update && apt-get install -y \
|
12 |
git ffmpeg curl unzip \
|
@@ -25,13 +18,13 @@ RUN pip install gdown av huggingface_hub
|
|
25 |
RUN pip install -r requirements.txt
|
26 |
RUN pip install --no-binary :all: av # optional: rebuild av from source
|
27 |
|
28 |
-
# Download ckpt weights from Hugging Face (private or public)
|
29 |
-
RUN python -c "import os; from huggingface_hub import snapshot_download; snapshot_download( repo_id='roll-ai/FloVD-weights', repo_type='dataset', local_dir='/app/ckpt', allow_patterns='ckpt/**', token=os.getenv('HF_TOKEN'))"
|
30 |
|
31 |
|
32 |
-
# Download example camera pose
|
33 |
-
RUN mkdir -p /weights/assets/manual_poses
|
34 |
-
RUN curl -L -o /weights/assets/manual_poses/example.txt https://huggingface.co/datasets/mutqa/FloVD-HF-Assets/resolve/main/example.txt
|
35 |
|
36 |
# Set permissions (optional but helpful)
|
37 |
RUN chmod -R 777 /app
|
|
|
1 |
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
|
2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
# Install OS-level dependencies
|
4 |
RUN apt-get update && apt-get install -y \
|
5 |
git ffmpeg curl unzip \
|
|
|
18 |
RUN pip install -r requirements.txt
|
19 |
RUN pip install --no-binary :all: av # optional: rebuild av from source
|
20 |
|
21 |
+
# # Download ckpt weights from Hugging Face (private or public)
|
22 |
+
# RUN python -c "import os; from huggingface_hub import snapshot_download; snapshot_download( repo_id='roll-ai/FloVD-weights', repo_type='dataset', local_dir='/app/ckpt', allow_patterns='ckpt/**', token=os.getenv('HF_TOKEN'))"
|
23 |
|
24 |
|
25 |
+
# # Download example camera pose
|
26 |
+
# RUN mkdir -p /weights/assets/manual_poses
|
27 |
+
# RUN curl -L -o /weights/assets/manual_poses/example.txt https://huggingface.co/datasets/mutqa/FloVD-HF-Assets/resolve/main/example.txt
|
28 |
|
29 |
# Set permissions (optional but helpful)
|
30 |
RUN chmod -R 777 /app
|