Commit ed4fa2a4 authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 0f12dd11
......@@ -56,16 +56,16 @@ source /opt/dtk/env.sh
## 环境配置
## 推理
### 设置环境变量
**模型下载**
下载链接:http://113.200.138.88:18080/aimodels/stablediffusion_v2.1_migraphx 下面运行示例命令中的onnx-model-path参数就设置为该下载后onnx模型目录。
**设置环境变量**
```shell
export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
```
### 安装依赖
**安装依赖**
```shell
# 进入python示例目录
cd <path_to_sd2.1_migraphx>
......@@ -73,12 +73,6 @@ cd <path_to_sd2.1_migraphx>
# 安装依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
```
### 模型下载
下载链接:http://113.200.138.88:18080/aimodels/stablediffusion_v2.1_migraphx 下面运行示例命令中的onnx-model-path参数就设置为该下载后onnx模型目录。
## 推理
### 运行示例
stablediffusion_v2.1模型的推理示例程序是Diffusion_test_offload_false.py,使用如下命令运行该推理示例:
......@@ -87,7 +81,7 @@ stablediffusion_v2.1模型的推理示例程序是Diffusion_test_offload_false.p
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
推理结果:
......@@ -96,7 +90,6 @@ python程序运行结束后,会在当前目录保存推理生成的图像。
<img src="./astro_horse.jpg" alt="Result" style="zoom: 50%;" />
### 精度
......
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