Fix docker cache image for dev (basetenlabs#1105)

* Change model copy dockerfile for improving cache
Distillative-AI · Aug 28, 2024 · aa2840f · aa2840f
1 parent 7a7e4ab
commit aa2840f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/truss/templates/server.Dockerfile.jinja b/truss/templates/server.Dockerfile.jinja
@@ -75,17 +75,19 @@ COPY ./{{config.data_dir}} /app/data
 {%- endif %}
 
 COPY ./server /app
-{%- if model_dir_exists %}
-COPY ./{{ config.model_module_dir }} /app/model
-{%- endif %}
+
 COPY ./config.yaml /app/config.yaml
     {%- if config.live_reload %}
 COPY ./control /control
 RUN python3 -m venv /control/.env \
     && /control/.env/bin/pip3 install -r /control/requirements.txt
     {%- endif %}
+{%- if model_dir_exists %}
+COPY ./{{ config.model_module_dir }} /app/model
+{%- endif %}
 {% endblock %}
 
+
 {% block run %}
     {%- if config.live_reload %}
 ENV HASH_TRUSS {{truss_hash}}

diff --git a/truss/test_data/server.Dockerfile b/truss/test_data/server.Dockerfile
@@ -34,8 +34,8 @@ WORKDIR $APP_HOME
 # Copy data before code for better caching
 COPY ./data /app/data
 COPY ./server /app
-COPY ./model /app/model
 COPY ./config.yaml /app/config.yaml
+COPY ./model /app/model
 
 COPY ./packages /packages