update Dockerfile and Readme

Dockerfile

@@ -1,8 +1,8 @@
-FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
+FROM mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
 
 WORKDIR /workspace
 ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
 
 RUN pip install diffusers==0.34.0
 RUN sed -i 's|source /torch/venv3/pytorch/bin/activate|source /torch/venv3/pytorch_infer/bin/activate|' /root/.bashrc
-COPY main.py test.sh dataset.json /workspace/
+COPY main.py dataset.json /workspace/

README.md

@@ -4,6 +4,7 @@
 ```bash
 docker build -t diffusers:v0.1 .
 ```
+其中基础镜像 mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 联系寒武纪厂商技术支持可获取
 
 ### 模型下载
 模型地址：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5

run_in_docker.sh

@@ -1,4 +1,4 @@
 #! /usr/bin/env bash
 image=diffusers:v0.1
 device=cambricon_dev1
-docker run -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=/dev/$device:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image python3 main.py --model "./stable-diffusion-v1-5" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
+docker run -v `pwd`:/workspace -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=/dev/$device:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image python3 main.py --model "./stable-diffusion-v1-5" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16