roll-ai commited on
Commit
a4e9b00
·
verified ·
1 Parent(s): c44240c

Update Dockerfile

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