forked from EngineX-Ascend/enginex-ascend-910-diffusers
add Dockerfile for mrv100
This commit is contained in:
@@ -3,5 +3,4 @@ FROM zibo.harbor.iluvatar.com.cn:30000/saas/bi100-3.2.1-x86-ubuntu20.04-py3.10-p
|
|||||||
WORKDIR /workspace
|
WORKDIR /workspace
|
||||||
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
|
ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1
|
||||||
RUN pip install diffusers==0.34.0
|
RUN pip install diffusers==0.34.0
|
||||||
COPY main.py /workspace/
|
COPY main.py test.sh dataset.json /workspace/
|
||||||
COPY test.sh /workspace/
|
|
||||||
|
|||||||
7
Dockerfile.mrv100
Normal file
7
Dockerfile.mrv100
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
FROM corex:4.3.0
|
||||||
|
|
||||||
|
WORKDIR /workspace
|
||||||
|
COPY whls-mrv100 /packages
|
||||||
|
RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
|
||||||
|
# RUN pip install /packages/*.whl
|
||||||
|
COPY main.py test.sh dataset.json /workspace/
|
||||||
3
run_in_docker_mrv100.sh
Executable file
3
run_in_docker_mrv100.sh
Executable file
@@ -0,0 +1,3 @@
|
|||||||
|
#! /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
|
||||||
Reference in New Issue
Block a user