如何通过谷歌云编辑器运行Stable Diffusion(sd)生成图片？

如果你的电脑没有独立显卡，想要跑一个本地化的图片生成模型比较复杂，如果只是偶尔用一下来测试学习，可以使用Google Colab 来调用免费资源生成图片。

如何通过谷歌云编辑器运行sd生成图片？

要通过谷歌云编辑器（Google Colab）运行 Stable Diffusion（SD）生成图片，你可以按照以下步骤操作：

1. 创建新的 Google Colab 笔记本

• 访问 Google Colab，点击“新建笔记本”。

2. 设置环境

在你的 Colab 笔记本中，首先安装所需的依赖包。运行以下代码：

!pip install torch torchvision torchaudio !pip install transformers !pip install diffusers !pip install accelerate !pip install pillow

!pip install torch torchvision torchaudio 
!pip install transformers 
!pip install diffusers 
!pip install accelerate 
!pip install pillow

3. 加载并配置模型

然后，加载 Stable Diffusion 模型。你可以选择 Hugging Face 上的模型，或者使用其他来源。from diffusers import StableDiffusionPipeline import torch # 需要 Hugging Face 认证 token pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v-1-4-original", torch_dtype=torch.float16) pipe = pipe.to("cuda") # 使用 GPU 加速（如果你选择 GPU 运行 Colab）

from diffusers import StableDiffusionPipeline import torch 

# 需要 Hugging Face 认证 token 

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v-1-4-original", torch_dtype=torch.float16) pipe = pipe.to("cuda") 

# 使用 GPU 加速（如果你选择 GPU 运行 Colab）

4. 生成图片

输入你想要生成的文本提示（Prompt），然后运行生成图片的代码。

prompt = "A fantasy landscape with mountains and rivers"

image = pipe(prompt).images[0]

# 显示图片

image.show()

# 保存图片

image.save("generated_image.png")

5. 下载生成的图片

如果你想下载图片到本地，执行以下代码：from google.colab import files files.download("generated_image.png")

小贴士：

• GPU 加速：在 Colab 中，默认情况下你可以选择启用 GPU（通过 编辑 > 笔记本设置 > 硬件加速器 选择 GPU），这样可以大大加速图像生成过程。
• Hugging Face 认证：如果你使用 Hugging Face 上的模型，可能需要一个认证 token。你可以从 Hugging Face 官网 注册并获取。

通过这些步骤，你就可以在 Google Colab 上运行 Stable Diffusion 生成图片了！

---

本教程为一系列AI开发部署实施测试模型等。大模型部署涉及的内容多而复杂，技术更新日益变化，如需交流可以加微信交流。我建了一个微信群用来交流各种AI使用部署问题。加微信：d11056888交流AI使用部署项目实战，我拉你进群交流AI使用和项目经验。