feat: 修改Dockerfile

Dockerfile

 FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
-RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang/photomaker-v2.git
-WORKDIR /root/photomaker-v2/PhotoMaker-V2
+ARG IMAGE=photomaker-v2
+ARG IMAGE_UPPER=PhotoMaker-V2
+ARG BRANCH=gpu
+RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
+WORKDIR /root/$IMAGE/$IMAGE_UPPER
 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/
 
@@ -8,12 +11,14 @@ RUN pip install -r requirements.txt && \
 # Prod  #
 #########
 FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
+ARG IMAGE=photomaker-v2
+ARG IMAGE_UPPER=PhotoMaker-V2
 COPY chenyh/photomaker/frpc_linux_amd64_v0.2 /opt/conda/lib/python3.10/site-packages/gradio/
 RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2
-COPY chenyh/photomaker-v2/TencentARC/PhotoMaker-V2 /root/PhotoMaker-V2/TencentARC/PhotoMaker-V2
-COPY chenyh/photomaker-v2/TencentARC/t2i-adapter-sketch-sdxl-1.0 /root/PhotoMaker-V2/TencentARC/t2i-adapter-sketch-sdxl-1.0
-COPY chenyh/photomaker-v2/SG161222/RealVisXL_V4.0 /root/PhotoMaker-V2/SG161222/RealVisXL_V4.0
+COPY chenyh/$IMAGE/TencentARC/PhotoMaker-V2 /root/$IMAGE_UPPER/TencentARC/PhotoMaker-V2
+COPY chenyh/$IMAGE/TencentARC/t2i-adapter-sketch-sdxl-1.0 /root/$IMAGE_UPPER/TencentARC/t2i-adapter-sketch-sdxl-1.0
+COPY chenyh/$IMAGE/SG161222/RealVisXL_V4.0 /root/$IMAGE_UPPER/SG161222/RealVisXL_V4.0
 COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
-COPY --from=base /root/photomaker/PhotoMaker /root/PhotoMaker
-COPY --from=base /root/photomaker/启动器.ipynb /root/photomaker/start.sh /root/
+COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
+COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE_UPPER/start.sh /root/
 #RUN python -c "import transformers; transformers.utils.move_cache()"
\ No newline at end of file