https://github.com/xserrat/docker-facebook-demucs/pull/11

xserrat · Nov 16, 2021 · e1b72d3 · e1b72d3
1 parent 971481f
commit e1b72d3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,7 @@
 FROM python:3.8
 
 USER root
+ENV TORCH_HOME=/data/models
 
 # Install Git
 RUN apt install git

diff --git a/Makefile b/Makefile
@@ -12,7 +12,6 @@ run: build ## Build & Run the demucs spliting the tracks placed in the input fol
 		-v $(current-dir)input:/data/input \
 		-v $(current-dir)output:/data/output \
 		-v $(current-dir)models:/data/models \
-		-e TORCH_HOME=/data/models \
 		facebook/demucs:latest \
 		"python3 -m demucs.separate --out /data/output \
 		/data/input/$(track)"