Stable Diffusion AI算图使用手册(2-4):ASRock AI QuickSet动手玩,最佳化AMD显示卡提升10倍算图效能

Stable Diffusion AI算图使用手册(2-4):ASRock AI QuickSet动手玩,最佳化AMD显示卡提升10倍算图效能

ASRock推出的AI QuickSet软件套件包含Stable Diffusion WebUI并支持通过Microsoft Olive加速算图效能,根据实测可以带来10倍效能表现。

安装与模型优化

先前AMD在官方博客张贴了在Stable Diffusion WebUI DirectML分枝版本的使用教程,可以通过Microsoft Olive工具将原本PyTorch格式的模型转换为ONNX格式,并在运算时使用DirectML API,达到接近10倍的效能表现。

不过笔者在跟随教程操作的过程中遇到一些技术问题,目前尚无法排除,导致无法套用效能优化,好在ASRock推出的AI QuickSet软件套件,大幅简化整体安装流程,用户只需在精灵的协助下安装程序,不需额外的设置就可以享受效能AI大幅提升的优势。

不过需要注意的是,电脑中需要有ASRock AMD Radeon RX 7000系列显卡才能安装AI QuickSet。 ASRock官方也表示目前AI QuickSet程序将会持续发展,并在未来加入更多实用的AI应用程序,为用户带来更多元便捷的AI功能。

AI QuickSet下载位置:
https://www.asrock.com/microsite/aiquickset/index.html

根据AMD官方提供的数据,使用Stable Diffusion WebUI DirectML分枝版本搭配转换为ONNX格式的模型,可以带来近10倍的算图效能表现。▲ 根据AMD官方提供的数据,使用Stable Diffusion WebUI DirectML分枝版本搭配转换为ONNX格式的模型,可以带来近10倍的算图效能表现。

ASRock推出的AI QuickSet软件套件可以简化Stable Diffusion WebUI DirectML分枝版本的安装手续。▲ ASRock推出的AI QuickSet软件套件可以简化Stable Diffusion WebUI DirectML分枝版本的安装手续。

这次测试使用的硬件为ASRock Radeon RX 7800XT Steel Legend显卡。▲ 这次测试使用的硬件为ASRock Radeon RX 7800XT Steel Legend显卡。

读者可以在ASRock官方网站下载AI QuickSet。▲ 读者可以在ASRock官方网站下载AI QuickSet。

安装手续与一般Windows应用程式相同。▲ 安装手续与一般Windows应用程序相同。

以1.1.13版为例,安装完成后,桌面会出现3个捷径图标。▲ 以1.1.13版为例,安装完成后,桌面会出现3个捷径图标。

出图效能提升10倍

如果想要使用优化性能的AI算图环境的话,可以执行「Launch Stable Diffusion WebUI ONNX」捷径。 不过需要注意的是,目前DirectML版本仅支持使用Stable Diffusion 1.5基础模型,网页界面中的Olive模型转换工具无法用于转换其他Checkpoint模型(程序以Hard coding方式指定模型文件),且尚不支持套LoRA,功能限制比较多。 若想要使用其他模型则可以执行「Launch Stable Diffusion WebUI」捷径。

开启DirectML版本的Stable Diffusion WebUI界面后,需在左上角的Stable Diffusion checkpoint下拉式选单选择「stable-diffusion-v1-5-olive [Optimized]」,其余操作则与一般版本相同。

笔者使用ASRock Radeon RX 7800XT Steel Legend进行测试,过程将Batch size分别设置为1、4,Batch count则固定为1,执行2轮测试确定测试结果无极端值后,取平均作为测试成绩。

从结果可以看到,AMD阵营显卡在优化之后,效能有接近在768 x 768分辨率下,Batch size为1时每次计算1张图片,DirectML版本的效能为一般版本的约4.64倍。 若将Batch size设置为4,每次同时计算4张图片,DirectML版本的速度不会受到太大的影响,但一般版本的速度却会严重下降,导致性能落差达到45.29倍。

第一次执行的时候,程序会自动安装需要的文件并转换模型,需要等待比较久的时间。▲ 第一次执行的时候,程序会自动安装需要的文件并转换模型,需要等待比较久的时间。

第二次执行之后开启的速度就会恢复正常。▲ 第二次执行之后开启的速度就会恢复正常。

执行时需在左上角的Stable Diffusion checkpoint下拉式菜单选择「stable-diffusion-v1-5-olive [Optimized]」,以套用最佳化模型。▲ 执行时需在左上角的Stable Diffusion checkpoint下拉式菜单选择「stable-diffusion-v1-5-olive [Optimized]」,以套用最佳化模型。

DirectML版本的参数设置与执行结果。▲ DirectML版本的参数设置与执行结果。

在Batch size设置为1时,DirectML版本的效能为一般版本的约4.64倍。▲ 在Batch size设置为1时,DirectML版本的效能为一般版本的约4.64倍。

AMD阵营的Olive工具可以带来有效的效能提升,但与NVIDIA提供的TensorRT最佳化工具相比方便性与实用度都稍嫌不足,同样有待日后更新改善。

(0)
老哥老哥

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注