"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "6c0e3972dcd80f08d4d30bab1bbd065ba0b658d1"
Commit cfd1a1dd authored by chenpangpang's avatar chenpangpang
Browse files

feat: 增加一个新模型;增加gradio依赖

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