如何在 Google Colab 上运行 stable-diffusion--large-turbo

心靈之曲
发布: 2024-12-09 08:03:02
转载
1171人浏览过

stable-diffusion-3.5-large-turbo 是一种高精度文本到图像模型。

本指南将解释如何在 google colab 上设置和运行模型。


先决条件

访问拥抱脸。

如何在 Google Colab 上运行 stable-diffusion--large-turbo

要使用 stable-diffusion-3.5-large-turbo,您需要一个 huggingface 帐户。

如果您还没有帐户,请创建一个帐户。

注册后,您将看到以下屏幕:

如何在 Google Colab 上运行 stable-diffusion--large-turbo

输入所需信息,您将立即获得模型的访问权限。

如果您想下载并使用该模型,您将需要访问令牌。从您的帐户页面创建一个:

如何在 Google Colab 上运行 stable-diffusion--large-turbo

通过右上角的个人资料图标导航到您的帐户页面,转到访问令牌选项卡,然后选择创建新令牌来创建令牌。


运行代码

安装所需的库

首先,在 google colab 中安装必要的库:

!pip install --quiet -u transformers
登录后复制

-u 选项将库更新到最新版本,--quiet 禁止下载消息。

讯飞智作-讯飞配音
讯飞智作-讯飞配音

讯飞智作是一款集AI配音、虚拟人视频生成、PPT生成视频、虚拟人定制等多功能的AI音视频生产平台。已广泛应用于媒体、教育、短视频等领域。

讯飞智作-讯飞配音 67
查看详情 讯飞智作-讯飞配音

验证您的帐户

通过运行以下命令并输入您之前创建的令牌来验证您的 huggingface 帐户:

!huggingface-cli login
登录后复制

下载模型

使用以下 python 代码加载并设置模型:

import torch
from diffusers import stablediffusion3pipeline

pipe = stablediffusion3pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
pipe = pipe.to("cuda")
登录后复制
注意:该模型消耗约 27gb 内存。

生成图像

通过运行此代码来生成图像来测试设置:

prompt = "a capybara holding a sign that reads hello fast world"
save_filename = "capybara.png"
image = pipe(
    prompt,
    num_inference_steps=4,
    guidance_scale=0.0,
).images[0]
登录后复制

您可以在 diffusers github 文档中找到这些参数的解释。

保存并显示生成的图像:

image.save(save_filename)
image
登录后复制

如何在 Google Colab 上运行 stable-diffusion--large-turbo

以上就是如何在 Google Colab 上运行 stable-diffusion--large-turbo的详细内容,更多请关注php中文网其它相关文章!

相关标签:
最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:dev.to网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号