开源绘图AI StableDiffusion现在可以在基于WebGPU的浏览器中运行

💡 原文中文,约700字,阅读约需2分钟。
📝

内容提要

StableDiffusion是一款流行的AI绘图项目,现在有开发者利用Google Chrome的WebGPU功能,搭建了可以在浏览器中运行的StableDiffusion。项目需要浏览器支持WebGPU和WebAssembly,下载大型模型文件,占用大量内存,生成图像速度较慢。不支持多线程,打开DevTools会减慢速度。项目地址:https://github.com/dakenf,演示站点:https://islamov.ai/stable-diffusion-webgpu/。

🎯

关键要点

  • StableDiffusion是一款流行的AI绘图项目,得益于开源,吸引了更多开发者参与。
  • 开发者利用Google Chrome的WebGPU功能,搭建了可以在浏览器中运行的StableDiffusion。
  • StableDiffusionPipeline已从Python移植到JS,但由于采用单线程运行,速度较慢。
  • 项目主要用于演示,运行条件较为苛刻,包括浏览器支持WebGPU和WebAssembly。
  • 需要在线下载3.5GB的模型文件,运行时占用8GB内存。
  • 每个步骤生成图像大约需要1分钟,且不支持多线程。
  • 打开DevTools会使运行速度减慢两倍,UNET在GPU上运行速度仅为10%。
  • 生成可接受的图像大约需要20步,保守估计需30分钟以上。
  • 项目地址为:https://github.com/dakenf,演示站点为:https://islamov.ai/stable-diffusion-webgpu/
➡️

继续阅读