pragnakalp
commited on
Commit
•
125c3bf
1
Parent(s):
881f8f5
Update Dockerfile
Browse files- 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(
|
20 |
-
RUN sed -i 's/.load(audio2pose)/.load(audio2pose,map_location=torch.device(
|
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 |
|