Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
PhotoMaker
Commits
576ceecb
Commit
576ceecb
authored
Jul 26, 2024
by
chenpangpang
Browse files
feat: 新增SG161222/RealVisXL_V4.0模型
parent
64d88f4c
Pipeline
#1423
canceled with stages
Changes
2
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
3 deletions
+6
-3
Dockerfile
Dockerfile
+4
-2
hf_down.py
hf_down.py
+2
-1
No files found.
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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment