pragnakalp commited on
Commit
d98f65b
1 Parent(s): e374e38

Update Dockerfile

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