本地部署!文生图LCM超简单教程

这是经过蒸馏的模型,所以 prompt 不支持中文(仅英文)

总共8G,比我之前那个5G的 Turbo 生成的质量简直好几百倍(起码不会怪异)

一、下载模型

魔塔社区-LCM

命令如下

python 复制代码
modelscope download --model q2792046875/lcm

注意模型下载到的路径:

之后可以把它移动到你喜欢的位置

二、使用模型

现在,假设你已经到了下载的模型的同级目录,也就是你现在在 lcm 这个文件夹旁边

在这里创建一个.py文件

python 复制代码
from diffusers import DiffusionPipeline, LCMScheduler
import torch

pipe = DiffusionPipeline.from_pretrained("lcm", scheduler = LCMScheduler.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", subfolder = "scheduler"))
pipe.to(torch_device="cuda", torch_dtype=torch.float32)

prompt = "Self-portrait oil painting, a beautiful cyborg with golden hair, 8k"

num_inference_steps = 4  # 推理步数
images = pipe(prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=8.0, lcm_origin_steps=50, output_type="pil").images[0]
images.save("generated_image.png")

推理步数支持59步以内,但是更建议1~8

(报错的话,把diffuser下载一下就行)

好了,玩去吧

相关推荐
美狐美颜SDK开放平台1 分钟前
多终端适配下的人脸美型方案:美颜SDK工程开发实践分享
人工智能·音视频·美颜sdk·直播美颜sdk·视频美颜sdk
哈__4 分钟前
CANN加速Image Captioning图像描述生成:视觉特征提取与文本生成优化
人工智能
觉醒大王7 分钟前
哪些文章会被我拒稿?
论文阅读·笔记·深度学习·考研·自然语言处理·html·学习方法
禁默8 分钟前
Ops-Transformer深入:CANN生态Transformer专用算子库赋能多模态生成效率跃迁
人工智能·深度学习·transformer·cann
杜子不疼.10 分钟前
基于CANN GE图引擎的深度学习模型编译与优化技术
人工智能·深度学习
L、21814 分钟前
深入理解CANN:面向AI加速的异构计算架构详解
人工智能·架构
chaser&upper20 分钟前
预见未来:在 AtomGit 解码 CANN ops-nn 的投机采样加速
人工智能·深度学习·神经网络
松☆24 分钟前
CANN与大模型推理:在边缘端高效运行7B参数语言模型的实践指南
人工智能·算法·语言模型
结局无敌30 分钟前
深度探究cann仓库下的infra:AI计算的底层基础设施底座
人工智能
m0_4665252930 分钟前
绿盟科技风云卫AI安全能力平台成果重磅发布
大数据·数据库·人工智能·安全