2023-03-28 11:09:31 +08:00
|
|
|
# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges.
|
|
|
|
|
|
|
|
|
|
FROM public.ecr.aws/iflytek-open/aiges-gpu:11.6-1.17-3.9.13-ubuntu1804-v2.0.0-rc6
|
|
|
|
|
|
2023-03-28 11:27:24 +08:00
|
|
|
WORKDIR /app
|
2023-03-28 11:09:31 +08:00
|
|
|
# do this if you are on the chinese server.
|
|
|
|
|
RUN pip3 config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
|
|
|
|
|
|
2023-03-28 11:27:24 +08:00
|
|
|
RUN mkdir -p requirements
|
|
|
|
|
|
|
|
|
|
ADD requirements.txt /requirements
|
|
|
|
|
|
|
|
|
|
RUN pip install -r /requirements/requirements.txt
|
2023-03-28 11:09:31 +08:00
|
|
|
|
2023-03-28 11:27:24 +08:00
|
|
|
RUN mkdir -p /app/model && \
|
|
|
|
|
wget https://github.com/AK391/animegan2-pytorch/zipball/main /app/model/
|
2023-03-28 11:09:31 +08:00
|
|
|
|
2023-03-28 11:27:24 +08:00
|
|
|
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
|
2023-03-28 11:09:31 +08:00
|
|
|
|
2023-03-28 11:27:24 +08:00
|
|
|
RUN wget https://github.com/bryandlee/animegan2-pytorch/raw/main/weights/face_paint_512_v1.pt /app/model/checkpoints/face_paint_512_v1.pt
|
