removing separate caching step (#872)

basetenlabs · Mar 21, 2024 · 4180705 · 4180705
1 parent eb430c1
commit 4180705
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 13 deletions.
diff --git a/truss/templates/cache.Dockerfile.jinja b/truss/templates/cache.Dockerfile.jinja
@@ -1,5 +1,3 @@
-FROM python:3.11-slim as cache_warmer
-
 RUN mkdir -p /app/model_cache
 WORKDIR /app
 

diff --git a/truss/templates/copy_cache_files.Dockerfile.jinja b/truss/templates/copy_cache_files.Dockerfile.jinja
diff --git a/truss/templates/server.Dockerfile.jinja b/truss/templates/server.Dockerfile.jinja
@@ -1,7 +1,3 @@
-{%- if model_cache %}
-{%- include "cache.Dockerfile.jinja" %}
-{%- endif %}
-
 {% extends "base.Dockerfile.jinja" %}
 
 {% block base_image_patch %}
@@ -51,10 +47,9 @@ RUN pip install -r {{server_requirements_filename}} --no-cache-dir && rm -rf /ro
 
 
 {% block app_copy %}
-{%- if model_cache %}
-# Copy data before code for better caching
-    {%- include "copy_cache_files.Dockerfile.jinja"%}
-{%- endif %}
+    {%- if model_cache %}
+    {%- include "cache.Dockerfile.jinja" %}
+    {%- endif %}
 
 {%- if external_data_files %}
 {% for url, dst in external_data_files %}