MariaKaiser commited on
Commit
595d7ce
·
verified ·
1 Parent(s): 347a5b5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -23,6 +23,9 @@ RUN pip install --no-cache-dir -r requirements.txt
23
  # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'vocab.json', cache_dir=MODEL_DIR); \
24
  # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'model.pth', cache_dir=MODEL_DIR)"
25
 
 
 
 
26
  # Copy the app code
27
  COPY . .
28
 
 
23
  # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'vocab.json', cache_dir=MODEL_DIR); \
24
  # hf_hub_download('MariaKaiser/EGTTS_finetuned', 'model.pth', cache_dir=MODEL_DIR)"
25
 
26
+ RUN pip install --no-cache-dir torch==2.2.2+cu121 torchaudio==2.2.2+cu121 \
27
+ --index-url https://download.pytorch.org/whl/cu121
28
+
29
  # Copy the app code
30
  COPY . .
31