diff --git a/Dockerfile b/Dockerfile index 80abc14..4ff36a1 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 +FROM 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 diff --git a/README.md b/README.md index e06a724..75f7025 100644 --- a/README.md +++ b/README.md @@ -4,6 +4,7 @@ ```bash docker build -t diffusers:v0.1 . ``` +其中基础镜像 mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 联系寒武纪厂商技术支持可获取 ### 模型下载 模型地址:https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5