Spaces:
Building
on
A10G
Building
on
A10G
Update Dockerfile
Browse files- Dockerfile +7 -3
Dockerfile
CHANGED
@@ -25,7 +25,11 @@ RUN apt-get update && \
|
|
25 |
unzip \
|
26 |
openjdk-8-jdk
|
27 |
|
28 |
-
|
|
|
|
|
|
|
|
|
29 |
RUN git clone https://github.com/CraftJarvis/ROCKET-1.git && \
|
30 |
cd ROCKET-1 && \
|
31 |
git pull && \
|
@@ -36,10 +40,10 @@ RUN git clone https://github.com/CraftJarvis/ROCKET-1.git && \
|
|
36 |
cd checkpoints && \
|
37 |
bash download_ckpts.sh
|
38 |
|
39 |
-
WORKDIR /app/ROCKET-1/rocket/stark_tech
|
40 |
RUN python -c "from huggingface_hub import hf_hub_download;hf_hub_download(repo_id='phython96/ROCKET-MCP-Reborn', filename='MCP-Reborn.zip', local_dir='.')" && \
|
41 |
unzip -q MCP-Reborn.zip && rm MCP-Reborn.zip && \
|
42 |
mkdir tmp
|
43 |
|
44 |
-
WORKDIR /app/ROCKET-1/rocket/arm
|
45 |
ENTRYPOINT ["python", "eval_rocket.py", "--port", "7860", "--sam-path", "/app/ROCKET-1/rocket/realtime_sam/checkpoints"]
|
|
|
25 |
unzip \
|
26 |
openjdk-8-jdk
|
27 |
|
28 |
+
RUN useradd -m -u 1000 user
|
29 |
+
USER user
|
30 |
+
ENV PATH="/home/user/.local/bin:$PATH"
|
31 |
+
|
32 |
+
WORKDIR $HOME/app
|
33 |
RUN git clone https://github.com/CraftJarvis/ROCKET-1.git && \
|
34 |
cd ROCKET-1 && \
|
35 |
git pull && \
|
|
|
40 |
cd checkpoints && \
|
41 |
bash download_ckpts.sh
|
42 |
|
43 |
+
WORKDIR $HOME/app/ROCKET-1/rocket/stark_tech
|
44 |
RUN python -c "from huggingface_hub import hf_hub_download;hf_hub_download(repo_id='phython96/ROCKET-MCP-Reborn', filename='MCP-Reborn.zip', local_dir='.')" && \
|
45 |
unzip -q MCP-Reborn.zip && rm MCP-Reborn.zip && \
|
46 |
mkdir tmp
|
47 |
|
48 |
+
WORKDIR $HOME/app/ROCKET-1/rocket/arm
|
49 |
ENTRYPOINT ["python", "eval_rocket.py", "--port", "7860", "--sam-path", "/app/ROCKET-1/rocket/realtime_sam/checkpoints"]
|