Spaces:
Paused
Paused
Update Dockerfile
Browse files- 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 |
-
|
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
|