diff --git a/Dockerfile b/Dockerfile index adcd842..47a6734 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,5 +3,4 @@ FROM zibo.harbor.iluvatar.com.cn:30000/saas/bi100-3.2.1-x86-ubuntu20.04-py3.10-p WORKDIR /workspace ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1 RUN pip install diffusers==0.34.0 -COPY main.py /workspace/ -COPY test.sh /workspace/ \ No newline at end of file +COPY main.py test.sh dataset.json /workspace/ diff --git a/Dockerfile.mrv100 b/Dockerfile.mrv100 new file mode 100644 index 0000000..c69d486 --- /dev/null +++ b/Dockerfile.mrv100 @@ -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/ diff --git a/run_in_docker_mrv100.sh b/run_in_docker_mrv100.sh new file mode 100755 index 0000000..11f9cc3 --- /dev/null +++ b/run_in_docker_mrv100.sh @@ -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