feat: 增加模型

Dockerfile

-FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
+FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel as base
 ARG IMAGE=instantmesh
 ARG IMAGE_UPPER=InstantMesh
 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 Ninja xformers triton
+RUN pip install Ninja triton && pip install xformers==0.0.24 --index-url https://download.pytorch.org/whl/cu121
 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
+FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel
 ARG IMAGE=instantmesh
 ARG IMAGE_UPPER=InstantMesh
+RUN apt-get update && apt-get install -y build-essential
 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/sudo-ai/zero123plus-v1.2 /root/$IMAGE_UPPER/sudo-ai/zero123plus-v1.2
-COPY chenyh/$IMAGE/TencentARC/InstantMesh /root/$IMAGE_UPPER/TencentARC/InstantMesh
+COPY chenyh/$IMAGE/TencentARC/InstantMesh/diffusion_pytorch_model.bin /root/$IMAGE_UPPER/TencentARC/InstantMesh/diffusion_pytorch_model.bin
+COPY chenyh/$IMAGE/TencentARC/InstantMesh/instant_mesh_large.ckpt /root/$IMAGE_UPPER/TencentARC/InstantMesh/instant_mesh_large.ckpt
+COPY chenyh/$IMAGE/facebook/dino-vitb16 /root/$IMAGE_UPPER/facebook/dino-vitb16
+COPY chenyh/$IMAGE/u2net.onnx /root/.u2net/u2net.onnx
 COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
+RUN echo 'export PATH=/opt/conda/lib/python3.10/site-packages/ninja/data/bin:$PATH' >> /root/.bashrc
 COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
 COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/

README.md

 # InstantMesh
 
+- https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx
\ No newline at end of file

hf_down.py

@@ -5,7 +5,8 @@ os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
 
 model_list = [
     "sudo-ai/zero123plus-v1.2",
-    "TencentARC/InstantMesh"
+    "TencentARC/InstantMesh",
+    "facebook/dino-vitb16"
 ]
 
 for model_path in model_list: