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

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -9
Dockerfile CHANGED
@@ -26,15 +26,8 @@ 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 "\
30
- from huggingface_hub import snapshot_download; \
31
- snapshot_download(\
32
- repo_id='roll-ai/FloVD-weights', \
33
- repo_type='dataset', \
34
- local_dir='/app/ckpt', \
35
- allow_patterns='ckpt/**', \
36
- token=os.getenv('HF_TOKEN')\
37
- )"
38
 
39
  # Download example camera pose
40
  RUN mkdir -p /weights/assets/manual_poses
 
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