Mac 部署 AIGC 图片生成服务——基于 stable-diffusion
AIGC 即人工智能内容生成,是目前非常火的一个概念。随着各种大模型的问世,通过 AI 来生成内容的能已经越来越强大。本文将从应用实践方面进行介绍如何在自己的 PC 电脑上部署一个强大的 AI 图片生成服务。
关于 AI 绘图,我相信你一定不太陌生,现在很多宣传图,插图其实都是有 AI 根据输入的文本信息来进行生成的。stable-diffusion 就是这样的一种图片生成 AI 系统,更重要的是他是开源的,通过它我们可以快速的部署自己的 AI 图片生成服务。
首先,我们可以从 Github 上下载一个开源的基于 stable-diffusion 的网页端应用程序,地址如下:
https://github.com/AUTOMATIC1111/stable-diffusion-webui
此服务的运行依赖 Python 环境,需要在个人电脑上安装 Python3.10.6 的版本。
之后需要安装 Pytorch,Pytorch 是一款基于 Python 的机器学习库,我们需要根据自己电脑的系统来选择安装命令,Pytorch 网址如下:
https://pytorch.org/get-started/locally/
如下图所示:
要验证 Pytorch 是否安装成功也很简单,直接进入 Python 可交互环境,输入:
如果没有任何报错,则说明已经安装成功,如下图所示:
现在我们还需要一个训练好的 AI 模型来指导生成效果,在如下网站中可以下载到很多训练好的模型:
选择一个感兴趣的模型,下载好的,将模型文件放入工程目录下的 models/Stable-diffusion 文件夹下面即可。
下面可以尝试运行下此 Web 项目,在工程目录下执行:
执行完成后,在浏览器打开如下地址:
页面如下所示:
可以看到,界面上左上角的位置可以选择要使用的模型,支持文字生成图片,图片生成图片,并可以进行图片生成的参数调整。下面可以尝试输入一个 prompt 来进行图片生成,需要注意,prompt 分为两部分,上面输入正向词汇,下面输入负向词汇。
需要注意,如果你无法访问https://civitai.com/网站,也可以尝试在https://huggingface.co/上面找一些训练好的生成图片的模型进行使用。
版权声明: 本文为 InfoQ 作者【珲少】的原创文章。
原文链接:【http://xie.infoq.cn/article/f51eb1512cb918afb75111b86】。文章转载请联系作者。
评论