Commit b04b3fa3 authored by yaoht's avatar yaoht
Browse files

update readme and docker image path

parent cfb6b1cd
......@@ -28,7 +28,7 @@ https://arxiv.org/pdf/2010.02502
拉取镜像:
```shell
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion-migraphx-centos7.6-dtk24.04.1-py310
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion-migraphx-ubuntu20.04-dtk24.04.3-py310
```
创建并启动容器:
......@@ -75,7 +75,7 @@ pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
```
### 模型下载
百度云下载 链接:https://pan.baidu.com/s/1JgJ0WUlhpey-ALsA48iyWw?pwd=mpp1 提取码:mpp1 下载后模型的目录为sd21-onnx,下面运行示例命令中的onnx-model-path参数就设置为该onnx模型目录。
下载链接:http://113.200.138.88:18080/aimodels/stablediffusion_v2.1_migraphx 下面运行示例命令中的onnx-model-path参数就设置为该下载后onnx模型目录。
## 推理
......@@ -84,7 +84,7 @@ pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
stablediffusion_v2.1模型的推理示例程序是Diffusion_test_offload_false.py,使用如下命令运行该推理示例:
```shell
python Diffusion_test_offload_false.py --prompt "a photograph of an astronaut riding a horse" --seed 13 --output astro_horse.jpg --steps 50 --fp16 all --onnx-model-path sd21-onnx
python Diffusion_test_offload_false.py --prompt "a photograph of an astronaut riding a horse" --seed 13 --output astro_horse.jpg --steps 50 --fp16 all --onnx-model-path stablediffusion_v2.1_migraphx
```
### result
......
FROM image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion-migraphx-centos7.6-dtk24.04.1-py310
FROM image.sourcefind.cn:5000/dcu/admin/base/custom:stablediffusion-migraphx-ubuntu20.04-dtk24.04.3-py310
RUN source /opt/dtk/env.sh
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment