fix: cleanup entrypoint

tinfoilsh · Jan 16, 2025 · c0faad1 · c0faad1
1 parent b144ef1
commit c0faad1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -7,6 +7,8 @@ RUN apt install -y iproute2
 
 COPY --from=shim /nitro-attestation-shim /nitro-attestation-shim
 
+ENV HOME=/
+
 RUN nohup bash -c "ollama serve &" && sleep 5 && ollama pull llama3.2:1b
 
-ENTRYPOINT ["sh", "-c", "echo Running && sleep 5 && /nitro-attestation-shim -d enclave-demo.tinfoil.sh -e [email protected] -u 11434 -c 7443 -l 443 -- /bin/ollama serve"]
+ENTRYPOINT ["/nitro-attestation-shim", "-s", "-d", "inference-enclave.tinfoil.sh", "-e", "[email protected]", "-u", "11434", "-c", "7443", "-l", "443", "--", "/bin/ollama", "serve"]