HF Space configuration

Dockerfile

@@ -42,14 +42,22 @@ WORKDIR /app
 RUN mix local.hex --force && \
     mix local.rebar --force
 
+
+# NOTE: Some config borrowed from this HF example:
+# https://huggingface.co/spaces/SpacesExamples/single_file_phx_bumblebee_ml/blob/main/Dockerfile
+
 # set build ENV
 ENV MIX_ENV="prod"
 
 # bumblebee - XLA_TARGET must exist before deps pulled
 ENV XLA_TARGET="cuda120"
-ENV BUMBLEBEE_CACHE_DIR="/data/cache/bumblebee"
+ENV BUMBLEBEE_CACHE_DIR="/app/.bumblebee"
 ENV XLA_CACHE_DIR="/data/cache/xla"
 
+ENV MIX_HOME="/app/.mix"
+ENV EXS_DRY_RUN="true"
+ENV MIX_INSTALL_DIR="/app/.mix"
+
 # install mix dependencies
 COPY mix.exs mix.lock ./
 RUN mix deps.get --only $MIX_ENV
@@ -127,14 +135,21 @@ ENV LC_ALL en_US.UTF-8
 WORKDIR "/app"
 RUN chown nobody /app
 
-# set runner ENV
+# NOTE: Some config borrowed from this HF example:
+# https://huggingface.co/spaces/SpacesExamples/single_file_phx_bumblebee_ml/blob/main/Dockerfile
+
+# set build ENV
 ENV MIX_ENV="prod"
 
-# Bumblebee
+# bumblebee - XLA_TARGET must exist before deps pulled
 ENV XLA_TARGET="cuda120"
-ENV BUMBLEBEE_CACHE_DIR="/data/cache/bumblebee"
+ENV BUMBLEBEE_CACHE_DIR="/app/.bumblebee"
 ENV XLA_CACHE_DIR="/data/cache/xla"
 
+ENV MIX_HOME="/app/.mix"
+ENV EXS_DRY_RUN="true"
+ENV MIX_INSTALL_DIR="/app/.mix"
+
 # ENV CUDA_HOME="/usr/local/cuda"
 # ENV PATH="${CUDA_HOME}/bin:${PATH}"
 # ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:$LD_LIBRARY_PATH"