sunjichen/enginex-ascend-910-diffusers
1
0
Fork 0
forked from EngineX-Ascend/enginex-ascend-910-diffusers
Code Pull Requests Activity

ascend

Browse Source
This commit is contained in:
Zhang Hao
2025-09-04 09:53:23 +08:00
parent 31c16831b1
commit 5d0b6f4e1f
12 changed files with 5 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@@ -1 +0,0 @@
Dockerfile.bi100

5
Dockerfile Normal file
View File

@@ -0,0 +1,5 @@
FROM quay.io/ascend/vllm-ascend:v0.10.0rc1
WORKDIR /workspace
RUN pip install diffusers==0.34.0
COPY main.py dataset.json /workspace/

5
Dockerfile.a100
View File

@@ -1,5 +0,0 @@
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
WORKDIR /workspace
# ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0 transformers sentencepiece
COPY main.py test.sh dataset.json /workspace/

5
Dockerfile.ascend
View File

@@ -1,5 +0,0 @@
FROM quay.io/ascend/vllm-ascend:v0.10.0rc1
WORKDIR /workspace
RUN pip install diffusers==0.34.0
COPY main.py dataset.json /workspace/

6
Dockerfile.bi100
View File

@@ -1,6 +0,0 @@
FROM git.modelhub.org.cn:980/enginex-iluvatar/bi100-3.2.1-x86-ubuntu20.04-py3.10-poc-llm-infer:v1.2.2
WORKDIR /workspace
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
RUN pip install diffusers==0.34.0
COPY main.py test.sh dataset.json /workspace/

8
Dockerfile.mlu370
View File

@@ -1,8 +0,0 @@
FROM git.modelhub.org.cn:9443/enginex-cambricon/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/

5
Dockerfile.mrv100
View File

@@ -1,5 +0,0 @@
FROM git.modelhub.org.cn:980/enginex-iluvatar/mr100_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/

3
run_in_docker_a100.sh
View File

@@ -1,3 +0,0 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:a100-0.2
docker run -it -v /home/zhanghao/workspace:/workspace -v /mnt:/mnt $image bash

4
run_in_docker_ascend.sh
View File

@@ -1,4 +0,0 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:ascend-0.2
device=1
docker run -it -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 --entrypoint bash $image

3
run_in_docker_bi100.sh
View File

@@ -1,3 +0,0 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:bi100-0.2
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=dev/iluvatar1:/dev/iluvatar0 $image bash

5
run_in_docker_mlu370.sh
View File

@@ -1,5 +0,0 @@
#! /usr/bin/env bash
# cnmon
image=harbor-contest.4pd.io/zhanghao/diffusers:mlu370-0.2
device_id=2
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=/dev/cambricon_dev$device_id:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image bash

3
run_in_docker_mrv100.sh
View File

@@ -1,3 +0,0 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:mrv100-0.2
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=dev/iluvatar0:/dev/iluvatar0 $image bash

1
test.sh
View File

@@ -1 +0,0 @@
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 --num_inference_steps 20