Add Boltz

RareCompute · Dec 3, 2024 · 5055342 · 5055342
1 parent f794a41
commit 5055342
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/apps/boltz/Dockerfile b/apps/boltz/Dockerfile
@@ -64,6 +64,7 @@ RUN \
   && rm -rf /root/.cache /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 COPY --chown=${UID}:${GID} ./apps/boltz/entrypoint.sh /entrypoint.sh
+RUN chmod -R 755 /entrypoint.sh
 
 USER ${USERNAME}
 WORKDIR /cache
@@ -72,4 +73,6 @@ VOLUME ["/cache"]
 WORKDIR /data
 VOLUME ["/data"]
 
+WORKDIR /app
+
 ENTRYPOINT ["/usr/bin/catatonit", "--", "/entrypoint.sh"]
diff --git a/apps/boltz/README.md b/apps/boltz/README.md
@@ -14,10 +14,9 @@ You can configure the docker image using the below environment variables
 
 | Environment Variable   | CLI Flag                 | Type                 | Default Value               | Description                                                         |
 | ---------------------- | ------------------------ | -------------------- | --------------------------- | ------------------------------------------------------------------- |
-| `SHELL`                | N/A                      | `FLAG`               | `False`                     | Start docker container as a shell                                   |
 | `USERNAME`             | N/A                      | `STR`                | `rare`                      | Username for the container                                          |
-| `UID`                  | N/A                      | `INTEGER`            | `900`                       | UID to use for the container                                        |
-| `GID`                  | N/A                      | `INTEGER`            | `900`                       | GID to use for the container                                        |
+| `UID`                  | N/A                      | `INTEGER`            | `900`                       | UID for the container                                               |
+| `GID`                  | N/A                      | `INTEGER`            | `900`                       | GID for the container                                               |
 | `DATA_PATH`            | `--out_dir`              | `PATH`               | `/data`                     | The path where predictions will be saved.                           |
 | `CACHE_PATH`           | `--cache`                | `PATH`               | `/cache`                    | The directory for downloading data and models.                      |
 | `DEVICE_TYPE`          | `--accelerator`          | `[gpu, cpu, tpu]`    | `gpu`                       | The type of accelerator to use for predictions.                     |