Merge pull request #2868 from vvsotnikov/master

feat(docker): mount models folder as a volume
bmaltais · Oct 3, 2024 · 54c0bee · 54c0bee
2 parents 3f1d414 + d89f0be
commit 54c0bee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.dockerignore b/.dockerignore
@@ -3,6 +3,7 @@ cudnn_windows/
 bitsandbytes_windows/
 bitsandbytes_windows_deprecated/
 dataset/
+models/
 __pycache__/
 venv/
 **/.hadolint.yml

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -20,6 +20,7 @@ services:
       - /tmp
     volumes:
       - /tmp/.X11-unix:/tmp/.X11-unix
+      - ./models:/app/models
       - ./dataset:/dataset
       - ./dataset/images:/app/data
       - ./dataset/logs:/app/logs