Merge pull request #7 from togethercomputer/addLocalConfig

Create local-cfg.yaml
togethercomputer · Jun 15, 2023 · ec4c0c3 · ec4c0c3
2 parents 1691982 + 083e560
commit ec4c0c3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -8,5 +8,6 @@ RUN wget https://together-distro-packages.s3.us-west-2.amazonaws.com/archlinux/x
 chmod +x /usr/local/bin/together-node
 COPY app app
 COPY requirements.txt requirements.txt
+COPY local-cfg.yaml /home/user/cfg.yaml
 RUN pip install -r ./requirements.txt
 WORKDIR /app
diff --git a/local-cfg.yaml b/local-cfg.yaml
@@ -0,0 +1,7 @@
+# Installed to /home/user/cfg.yaml in container
+worker:
+  mode: local-service
+  group:
+    alloc: each
+  command: /usr/bin/python3 /app/serving.py
+  env: HF_HOME=/home/user