"vscode:/vscode.git/clone" did not exist on "11da4f17b7d704ee7ae8de4f8d3a3599bc05f4f6"
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 ...@@ -28,7 +28,7 @@ https://arxiv.org/pdf/2010.02502
拉取镜像: 拉取镜像:
```shell ```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 ...@@ -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 ...@@ -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,使用如下命令运行该推理示例: stablediffusion_v2.1模型的推理示例程序是Diffusion_test_offload_false.py,使用如下命令运行该推理示例:
```shell ```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 ### 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 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