五、AI测试:Stable Diffusion文生图效率将近翻倍
Stable Diffusion是一个基于图像的AI模型,只需要输入一句话、或者特定的关键词,就能生成一幅想要的图像,也是时下最火的AI生成画图工具。
2023年9月,NVIDIA推出了用于Stable Diffusion Web UI 的TensorRT 加速插件,能让具备Tensor单元的GeForce显卡在Stable Diffusion Web UI 中获得翻倍的性能。
本次我们将对比SDXL加速方法与PyTorch xFormers(常规加速方法)的性能差异。
在测试时,我们的采样方法选择DPM++ 2S a Karras,迭代步数为70,图片宽度和高度为854×1264,对比调用SDXL加速方法与PyTorch xFormers(常规加速方法)的性能差异。
使用PyTorch xFormers(常规加速方法)的生成图片共耗时7分钟44秒,而使用SDXL加速方法生成图片仅用时3分钟14秒,领先高达41%,提升了将近一半的效率。