pragnakalp commited on
Commit
125c3bf
1 Parent(s): 881f8f5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -16,8 +16,8 @@ RUN git clone https://huggingface.co/camenduru/pocketsphinx-20.04-t4 pocketsphin
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
19
- RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location=torch.device('cpu')) /' one-shot-talking-face/tools/interface.py
20
- RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device('cpu')) /' one-shot-talking-face/tools/interface.py
21
  RUN cat one-shot-talking-face/tools/interface.py
22
  RUN mkdir /content/out
23
 
 
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
19
+ RUN sed -i 's/.load(checkpoint_path)/.load(checkpoint_path,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
20
+ RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device("cpu")) /' one-shot-talking-face/tools/interface.py
21
  RUN cat one-shot-talking-face/tools/interface.py
22
  RUN mkdir /content/out
23