Commit 653cf5f3 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 去除tranformers需要移除cache的提示

parent 2fd02f85
......@@ -4,7 +4,7 @@ RUN cd /root && git clone -b gpu http://developer.hpccube.com/codes/chenpangpang
WORKDIR /root/photomaker/PhotoMaker
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
# PhotoMaker-base clone from https://github.com/TencentARC/PhotoMaker.git and change pipeline.py to spaces' pipeline.py
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/SG161222/RealVisXL_V4.0 /root/photomaker/PhotoMaker/SG161222/RealVisXL_V4.0
......@@ -18,3 +18,4 @@ RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.
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/
RUN python -c "import transformers; transformers.utils.move_cache()"
\ 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