pytorch 在Graphcore伊普上运行稳定扩散

ercv8c1e  于 2022-11-23  发布在  其他
关注(0)|答案(1)|浏览(167)

我一直在寻找一个版本的稳定扩散,这将能够运行在伊普的。目前(由于高可用性)到目前为止,我可以找到基于CUDA的只有。
现在我想知道是否有一种方法可以运行基于伊普的Cuda脚本/培训师/学习等?例如,在两者之间的翻译程序。
我怀疑有,我敢打赌,因为我找不到一个伊普的版本,我将不得不修改脚本:(。

ldioqlga

ldioqlga1#

有一个HuggingFace optimum库,它作为转换器在IPU上运行的互操作层,目前支持text2img。
对于库中不支持的其他模型,有一个指南介绍如何修改脚本,使其与IPU兼容here

相关问题