添加模型结果

Dockerfile

@@ -1,4 +1,4 @@
-FROM quay.io/ascend/vllm-ascend:v0.10.0rc1
+FROM git.modelhub.org.cn:9443/enginex-ascend/vllm-ascend:v0.10.0rc1
 
 WORKDIR /workspace
 RUN pip install diffusers==0.34.0

README.md

@@ -16,10 +16,30 @@ docker build -t diffusers:v0.1 .
 ./run_in_docker.sh
 ```
 
-## 测试结果
-### stable-diffusion-v1-5模型
+## 测试结果对比（A100 vs 昇腾910B）
 
-|      | A100 平均生成时间（秒） | 昇腾910B 平均生成时间（秒） |
+### 数据集
+数据集如下：
+```json
+[
+    "A futuristic city skyline at sunset, with flying cars and neon lights, in cyberpunk style",
+    "A traditional Chinese courtyard covered in snow, with red lanterns glowing warmly",
+    "A portrait of a young woman in Renaissance style, oil painting with dramatic lighting",
+    "A cute baby panda playing with bamboo in a watercolor illustration style",
+    "A majestic dragon flying over mountains, depicted in traditional ink wash painting",
+    "A cozy reading corner with bookshelves, a cat on the sofa, and sunlight streaming through the window",
+    "A knight in silver armor riding a horse, fantasy concept art with epic background",
+    "A bowl of ramen with detailed toppings, hyper-realistic food photography style",
+    "An astronaut floating in space, reflected in the helmet visor is planet Earth",
+    "A serene lake surrounded by autumn trees, painted in impressionist style"
+]
+```
+### 测试结果
+
+|  模型    | A100 平均生成时间（秒） | 昇腾910B 平均生成时间（秒） |
 |------|-------------------------|----------------------------|
-| 时间 | 1.6208                  | 7.2901                     |
+| AI-ModelScope/stable-diffusion-v1-5 | 1.6208                  | 7.2901                     |
+| zhanghaohit/karlo-v1-alpha | 3.1624                  | 5.2920                     |
+| AI-ModelScope/stable-diffusion-3.5-medium | 5.0231                  | 8.9607                     |
+| stabilityai/stable-diffusion-3-medium-diffusers	 | 3.7976                  | 23.5929                    |
 

run_in_docker.sh

@@ -1,3 +1,4 @@
 #! /usr/bin/env bash
 image=diffusers:v0.1
-docker run -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=dev/iluvatar1:/dev/iluvatar0 $image python3 main.py --model "./stable-diffusion-v1-5" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
+device=1
+docker run  -v `pwd`:/host -e ASCEND_VISIBLE_DEVICES=1 --device /dev/davinci$device:/dev/davinci0 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc -v /mnt:/mnt -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info -v /etc/ascend_install.info:/etc/ascend_install.info --privileged $image python3 main.py --model "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5" --json  "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16