Feature: Added volume persistence for models

Changes: - Volume for model caching
DennisJensen95 · Aug 9, 2023 · 3eba517 · 3eba517
1 parent a446657
commit 3eba517
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -43,7 +43,7 @@ COPY --from=BUILD /app/build-app /app/build-app
 
 EXPOSE 8080
 
+ENV RUSTBERT_CACHE=/data/rustbert_cache
 ENV RUST_LOG=info
-ENV LIBTORCH=/usr/local/lib/python3.7/dist-packages/torch
 
 ENTRYPOINT [ "./backend-server" ]
diff --git a/fly.toml b/fly.toml
@@ -15,3 +15,7 @@ auto_stop_machines = true
 auto_start_machines = true
 min_machines_running = 0
 processes = ["app"]
+
+[mount]
+source = "model_cache"
+destination = "/data"