aicloud
/
animeganv2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update dockerfile

This commit is contained in:
SOULOFCINDER 2023-03-28 11:27:24 +08:00
parent 9ff1cc3a24
commit d219e8dac8
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
Dockerfile
View File

@ -2,15 +2,20 @@
FROM public.ecr.aws/iflytek-open/aiges-gpu:11.6-1.17-3.9.13-ubuntu1804-v2.0.0-rc6
WORKDIR /app
# do this if you are on the chinese server.
RUN pip3 config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
ADD requirements.txt /home/aiges
RUN mkdir -p requirements
RUN pip install -r /home/aiges/requirements.txt
ADD requirements.txt /requirements
ADD wrapper /home/aiges
RUN pip install -r /requirements/requirements.txt
# companion.xfyun.iflytek:6868 is the default config center. It depends on how your athena serving framework built
CMD ["sh", "-c", "./AIservice -m=0 -c=aiges.toml -s=svcName -u=http://companion.xfyun.iflytek:6868 -p=AIaaS -g=dx"]
RUN mkdir -p /app/model && \
wget https://github.com/AK391/animegan2-pytorch/zipball/main /app/model/
RUN mkdir -p /app/model/checkpoints && \
wget https://github.com/bryandlee/animegan2-pytorch/raw/main/weights/face_paint_512_v2.pt /app/model/checkpoints/face_paint_512_v2.pt
RUN wget https://github.com/bryandlee/animegan2-pytorch/raw/main/weights/face_paint_512_v1.pt /app/model/checkpoints/face_paint_512_v1.pt