当前位置: 首页>后端>正文

国产的AI扩散模型Stable Diffusion来了

这里的国产AI扩散模型,是基于大名鼎鼎的Stable Diffusion,利用国产框架fork而来。这个开源的项目是用OneFlow深度学习框架,后者属于分布式性能(高效性)是深度学习框架的核心技术难点,围绕性能提升和异构分布式扩展,秉持静态编译和流式并行的核心理念和架构,解决了集群层面的内存墙挑战。

国产的AI扩散模型Stable Diffusion来了,第1张

最近疲于奔命,也没跟进学习深度学习相关技术和发展,所以对扩散模型属于一知半解。故而机智客只是大致看了一下GitHub的项目介绍,不求甚解。虽然是国产项目,但这个并不是突出中文,而是突出效率。被成为全球最快的Stable Diffusion。

要运行这个项目,可以用Docker也可以不用Docker而直接安装搭配。而不用Docker则要一步步来,先安装OneFlow,再Clone和安装OneFlow的fork diffusers。然后是python3 -m pip install -e .[oneflow],再就是git clone https://github.com/Oneflow-Inc/transformers.git 安装fork的transformers之后才是python3 -m pip install -e .总之有点拖拉。

项目页还列举了一个运行的示范代码,是在fp16版下的。看代码的意思是载入必要库,pipe加载必要的预训练网络,然后用CUDA,以一句示例句子a photo of an astronaut riding a horse on mars来生成图像,遍历后保存导出png图像的。说是速度快,有条件的朋友可以试试。当然这里没注意预训练网络模型是不是比较方便弄,反正上GitHub众所周知死慢死慢的。


https://www.xamrdz.com/backend/36b1942634.html

相关文章: