lengyue233 commited on
Commit
129c2fc
1 Parent(s): 07e9572

update torch path for demucs

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -22,5 +22,6 @@ ENV GRADIO_SERVER_PORT=7860
22
  ENV NUMBA_CACHE_DIR=/tmp/numba
23
  ENV TRANSFORMERS_CACHE=/tmp/huggingface
24
  ENV MPLCONFIGDIR=/tmp/matplotlib
 
25
 
26
  CMD python3 tools/hifisinger/inference.py --config configs/exp_hifisinger.py --checkpoint checkpoints/checkpoint.ckpt --speaker opencpop --max_slice_duration 120 --min_silence_duration 2 --gradio
 
22
  ENV NUMBA_CACHE_DIR=/tmp/numba
23
  ENV TRANSFORMERS_CACHE=/tmp/huggingface
24
  ENV MPLCONFIGDIR=/tmp/matplotlib
25
+ ENV TORCH_HOME=/tmp/torch
26
 
27
  CMD python3 tools/hifisinger/inference.py --config configs/exp_hifisinger.py --checkpoint checkpoints/checkpoint.ckpt --speaker opencpop --max_slice_duration 120 --min_silence_duration 2 --gradio