Skip to content

GitLab

"test/tests.cpp" did not exist on "e7f1ca7fb1833bb3358f4b40b023e77622ebfaf4"
Commit 576ceecb authored by chenpangpang's avatar chenpangpang
Browse files

feat: 新增SG161222/RealVisXL_V4.0模型

parent 64d88f4c
Pipeline #1423 canceled with stages
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
Dockerfile
View file @ 576ceecb
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
RUN apt-get update && apt-get install -y gcc g++
RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker.git
WORKDIR /root/photomaker/PhotoMaker WORKDIR /root/photomaker/PhotoMaker
RUN apt-get update && apt-get install -y gcc g++
RUN pip install -r requirements.txt && \ RUN pip install -r requirements.txt && \
pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
RUN pip install git+https://github.com/TencentARC/PhotoMaker.git # PhotoMaker-base clone from https://github.com/TencentARC/PhotoMaker.git
RUN pip install git+http://developer.hpccube.com/codes/chenpangpang/PhotoMaker-base.git
COPY chenyh/photomaker/TencentARC/PhotoMaker /root/photomaker/PhotoMaker/TencentARC/PhotoMaker COPY chenyh/photomaker/TencentARC/PhotoMaker /root/photomaker/PhotoMaker/TencentARC/PhotoMaker
COPY chenyh/photomaker/SG161222/RealVisXL_V4.0 /root/photomaker/PhotoMaker/SG161222/RealVisXL_V4.0
######### #########
# Prod # # Prod #
......
hf_down.py
View file @ 576ceecb
...@@ -6,7 +6,8 @@ import json ...@@ -6,7 +6,8 @@ import json
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com' os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
model_list = [ model_list = [
"TencentARC/PhotoMaker" "TencentARC/PhotoMaker",
"SG161222/RealVisXL_V4.0"
] ]
for model_path in model_list: for model_path in model_list:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment