Commit 5cea14f1 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 增加模型,图片;增加下载模型代码

parent 133c21e2
......@@ -19,9 +19,9 @@ RUN mv /root/$IMAGE/$IMAGE_UPPER /root/ComfyUI/custom_nodes/$IMAGE_UPPER && \
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
ARG IMAGE=comfyui-photomaker-zho
ARG IMAGE_UPPER=ComfyUI-PhotoMaker-ZHO
COPY chenyh/$IMAGE/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/$IMAGE/coqui/XTTS-v2 /root/.local/share/tts/tts_models--multilingual--multi-dataset--xtts_v2
COPY chenyh/$IMAGE/SG161222/RealVisXL_V3.0 /root/ComfyUI/SG161222/RealVisXL_V3.0
COPY chenyh/$IMAGE/TencentARC/PhotoMaker/photomaker-v1.bin /root/ComfyUI/photomaker-v1.bin
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/ComfyUI /root/ComfyUI
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/
\ No newline at end of file
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/
COPY --from=base /root/$IMAGE/xiaozhan.jpeg /root/ComfyUI/input/
\ No newline at end of file
# pip install huggingface-cli
import os
import requests
import json
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
model_list = [
"SG161222/RealVisXL_V3.0"
]
for model_path in model_list:
os.system(f"huggingface-cli download --resume-download {model_path} --local-dir ./{model_path} --local-dir-use-symlinks False")
\ No newline at end of file
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