Compare commits

...

5 Commits

Author SHA1 Message Date
dog
ada13b390d Update README.md (#1)
Reviewed-on: https://git.modelhub.org.cn:9443/EngineX-Iluvatar/enginex-mr_series-diffusers/pulls/1
2025-12-04 15:38:53 +08:00
zhousha
4366128139 update Dockerfile 2025-09-15 17:44:24 +08:00
455848137a mount current dir to workspace 2025-08-28 17:51:11 +08:00
faf81388ba 更新 Dockerfile 2025-08-28 17:29:41 +08:00
e178e4de4a 更新 Dockerfile 2025-08-28 17:28:23 +08:00
3 changed files with 23 additions and 6 deletions

View File

@@ -1,5 +1,5 @@
FROM git.modelhub.org.cn:980/enginex-iluvatar/mr100_corex:4.3.0
FROM corex:4.3.0
WORKDIR /workspace
RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
COPY main.py test.sh dataset.json /workspace/
COPY main.py dataset.json /workspace/

View File

@@ -4,6 +4,7 @@
```bash
docker build -t diffusers:v0.1 .
```
其中,基础镜像 corex:4.3.0 通过联系天数智芯智铠100厂商技术支持可获取
### 模型下载
模型地址https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
@@ -17,7 +18,23 @@ docker build -t diffusers:v0.1 .
```
## 测试结果
| | A100 平均生成时间(秒) | 智铠100 平均生成时间(秒) |
|------|-------------------------|----------------------------|
| 时间 | 1.5 | 4.1 |
| 模型名称 | A100 平均生成时间(秒) | 智铠100 平均生成时间(秒) | 备注 |
| ----- | ----- | ----- | ----- |
| AI-ModelScope/stable-diffusion-3.5-medium | 5.3253 | 56.6237 | |
| AI-ModelScope/stable-diffusion-v1-5 | 1.3944 | 3.9280 | |
| Cleaner/lo_dress_gothic_style2_v2 | 1.8821 | 2.5121 | |
| Cleaner/Tyndalleffect_Light | 1.7133 | 2.5138 | |
| HanYixuan1/Retro_style | 1.6764 | 2.4901 | |
| JeffVan/Oil_Paint_Style_LORA | 1.6673 | 2.7132 | |
| lljjcc/IndianSarres | 1.7252 | 2.5209 | |
| lljjcc/outdoor | 1.6684 | 2.5230 | |
| muse/flux_vae | 1.0812 | 2.5350 | |
| MusePublic/489_ckpt_FLUX_1/base_model | 1.2286 | 2.5382 | |
| rewfueranro/cartoon_lora | 1.6880 | 2.6043 | |
| YorickHe/fairy_lora | 1.7765 | 2.5139 | |
| YorickHe/JK_uniform_lora | 1.7844 | 2.5219 | |
| YorickHe/outdoor_photo_lora | 1.6441 | 2.4851 | |
| YorickHe/polaroid_lora | 1.6272 | 2.5117 | |
| YorickHe/Winter_hanfu_lora | 1.5962 | 2.4892 | |
| ZackWang123/filmvelvia_lora | 1.8465 | 2.5157 | |

View File

@@ -1,3 +1,3 @@
#! /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
docker run -v `pwd`:/workspace -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