feat: 修改下载模型PhotoMaker-V2

602b5b91 · chenpangpang · 42ea6dbf · 602b5b91 · 602b5b91
Commit 602b5b91 authored Jul 29, 2024 by chenpangpang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Dockerfile Dockerfile +1 -1

hf_down.py hf_down.py +1 -1

No files found.
--- a/Dockerfile
+++ b/Dockerfile
@@ -6,7 +6,7 @@ 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/
 # 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-V2 /root/photomaker/PhotoMaker/TencentARC/PhotoMaker-V2
 COPY chenyh/photomaker/SG161222/RealVisXL_V4.0 /root/photomaker/PhotoMaker/SG161222/RealVisXL_V4.0

 #########

--- a/hf_down.py
+++ b/hf_down.py
@@ -6,7 +6,7 @@ import json
 os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

 model_list = [
-        "TencentARC/PhotoMaker",
+        "TencentARC/PhotoMaker-V2",
        "SG161222/RealVisXL_V4.0"
 ]