pragnakalp commited on
Commit
0dba002
1 Parent(s): aac73eb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -12,7 +12,7 @@ RUN pip install cmake==3.24.1.1
12
 
13
  RUN git lfs install
14
  RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
15
-
16
  RUN git clone https://huggingface.co/camenduru/one-shot-talking-face-20.04-t4 one-shot-talking-face && cd one-shot-talking-face && pip install -r requirements.txt && chmod 755 OpenFace/FeatureExtraction
17
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/test_script.py
18
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py
 
12
 
13
  RUN git lfs install
14
  RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphinx && cd pocketsphinx && cmake -S . -B build && cmake --build build --target install
15
+ RUN torch.device('cuda' if torch.cuda.is_available() else 'cpu')
16
  RUN git clone https://huggingface.co/camenduru/one-shot-talking-face-20.04-t4 one-shot-talking-face && cd one-shot-talking-face && pip install -r requirements.txt && chmod 755 OpenFace/FeatureExtraction
17
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/test_script.py
18
  RUN sed -i 's/.cuda()/ /' one-shot-talking-face/tools/interface.py