如何在 Google Colab 上运行稳定的扩散

Rifat WordPress Tutorials May 11, 2023

Stable Diffusion是一种强大的生成模型,已在大型图像和视频数据集上进行训练,以创建逼真的高质量样本。 它通过迭代优化噪声输入信号来生成细节和连贯性水平不断提高的图像。 该技术已广泛应用于 AI 生成艺术领域,让艺术家和爱好者能够创作出独特而迷人的艺术品。

另一方面,Google Colab 是一个基于云的平台,可以免费访问强大的 GPU,使其成为试验机器学习模型的绝佳选择。 它还允许用户轻松地与他人协作和共享他们的工作。 通过将 Stable Diffusion 与 Google Colab 相结合,我们可以访问预训练模型并以最少的设置和成本生成我们自己的 AI 艺术。

在本教程中,我们将引导您完成在 Google Colab 中开始使用 Stable Diffusion 的步骤,包括设置环境、加载预训练模型和生成 AI 艺术样本。 无论您是 AI 生成艺术领域的初学者还是经验丰富的从业者,本教程都将为您提供必要的知识,以开始创作您自己独特的艺术作品。

稳定扩散入门

首先,我们将简要演示在 Google Colab 上运行 Stable Diffusion 的整个过程,然后生成我们的初始图像集。

创建令人惊叹的网站

使用最好的免费页面构建器 Elementor

现在开始

抱脸账号

要使用 Stable Diffusion,我们需要从 Hugging Face 下载它,这是一个提供一系列模型和有用功能的存储库。 要下载 Stable Diffusion,我们首先需要使用 Hugging Face 创建一个帐户。 模型本质上是一个可以学习独立执行某些任务的程序。

使用 Hugging Face 创建帐户是一个简单的过程,包括访问他们在Hugging Face 的网站并遵循标准帐户创建程序。 完成注册后,您将收到一封确认电子邮件以验证您的帐户。 确认帐户后,您将能够下载 Stable Diffusion 并开始在 Google Colab 中试验 AI 生成的艺术作品。

将稳定的 Diffusion Colab 笔记本复制到 Google 云端硬盘

首先,类似于编辑其他人创建的任何 Google 文档,您需要访问Google Collab上的 Stable Diffusion Google Colab 笔记本,然后选择“文件”选项,然后选择“在云端硬盘中保存副本”。

这将打开一个新选项卡,其中笔记本已保存到您的驱动器中,其名称将是“Copy of Stable Diffusion with diffusers”。 随意根据您的喜好重命名它。

使用 GPU

在 Google Colab 中运行 Stable Diffusion 之前,务必确保笔记本配置为使用 GPU。 GPU 是高效处理大型数据集和执行涉及训练生成模型(如 Stable Diffusion)的复杂计算的重要组件。

要检查笔记本是否配置为使用 GPU,请转到“运行时”菜单并选择“更改运行时类型”。 将出现一个带有“硬件加速器”下拉菜单的小窗口。 确保选择“GPU”作为硬件加速器。

选择“GPU”后,单击“保存”以应用更改。 现在,笔记本配置为使用 GPU,这将使我们能够流畅高效地运行 Stable Diffusion。 通过此设置,我们现在可以继续加载预训练模型并生成我们自己的 AI 艺术。

运行第一个单元格

我们现在可以执行 Stable Diffusion Colab 笔记本中的初始单元。 为此,只需将鼠标悬停在每个单元格上,就会出现一个播放按钮。 单击播放按钮并等待单元格执行完毕。 当出现绿色复选标记时,您将知道单元格已完成其任务。

每个单元格上方都附有说明,解释其用途和功能。 如果您遇到诸如“此笔记本需要高 RAM”之类的消息,只需单击“确定”即可继续。

成功执行初始单元后,您应该准备好开始使用 Stable Diffusion。 该笔记本提供了一个用户友好的界面,用于加载预训练模型和生成 AI 生成的艺术作品。 无论您是该领域的新手还是经验丰富的从业者,本笔记本都可以作为探索稳定扩散在生成高质量和独特艺术品方面的可能性的宝贵工具。

运行第五个单元格以下载所需文件

我们将继续执行位于“稳定扩散管道”部分下的第五个单元格,它将下载所需的组件。

此外,我们还需要执行以下单元格。

pipe = pipe.to("cuda")

生成图像

在随后的单元格中,您可能已经注意到出现了一个图像。 这是我们将生成第一张图像的地方。 为此,只需在引号内输入一个文本字符串并执行该单元格。 作为说明,我提供了一个示例,其中我输入了以下文本:

a protoss cityscape with advanced technology, inspired by the game starcraft, making heavy use of light and shadow to create a sense of mystery and foreboding. the city sprawling below is a mix of organic and inorganic, with swirling energy currents and strange crystalline structures, illustrated in a realistic and detailed style by wei wang, artstationa protoss cityscape with advanced technology, inspired by the game starcraft, making heavy use of light and shadow to create a sense of mystery and foreboding. the city sprawling below is a mix of organic and inorganic, with swirling energy currents and strange crystalline structures, illustrated in a realistic and detailed style by wei wang, artstation

做得好! 生成该图像的过程应该不到一分钟。

包起来

恭喜您在 Google Colab 上使用来自 Hugging Face 的 Stable Diffusion 生成了您的第一张图像! 这只是您进入令人兴奋的 AI 生成艺术世界之旅的开始。

通过遵循本指南,我们希望我们已经帮助您克服了最初的障碍,并为您提供了必要的知识以开始进一步探索 Stable Diffusion。 我们鼓励您继续探索 Google Colab notebook 中的其他说明,这将允许您根据您的特定需求自定义和微调您的模型。

但是,我们知道您在此过程中可能会遇到一些问题或有疑问。 不要犹豫,发表评论,我们的团队会尽快回复您。 我们随时为您提供支持,帮助您在 AI 生成的艺术之旅中取得成功。 继续试验,我们迫不及待地想看看你接下来会创造什么!

Divi WordPress Theme