本地部署!文生图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下载一下就行)

好了,玩去吧

相关推荐
梦想的颜色1 分钟前
AI Agent 可观测性:破解多步推理黑盒|从概念、架构、指标到生产落地图鉴
人工智能·ai agent 可观测性·llm 追踪 trace·llmops 硬核实战·agent 生产运维
IT智慧客07314 分钟前
Harness 到底指什么
人工智能
大模型任我行12 分钟前
阿里:端到端文档解析模型OvisOCR2
人工智能·语言模型·自然语言处理·论文笔记
aixingkong92113 分钟前
Atlas 950 1024卡SuperPoD推测分析
网络·人工智能·硬件架构·硬件工程
今心上13 分钟前
关于d2l中train_ch3以及softmax中图在pycharm中显示不出来的解决方案
深度学习·机器学习·pycharm
触底反弹15 分钟前
🤯 面试被问 AI Workflow 和 Agent 有啥区别?3 张图 + 2 段代码讲清楚!
人工智能·设计模式·面试
步步精BBJconn17 分钟前
从GPU服务器到数据中心:AI服务器高压连接器的应用与发展趋势
大数据·运维·服务器·人工智能·科技·物联网
K姐研究社23 分钟前
Codex 怎么用国产模型?Kimi + CC Switch 接入配置指南
人工智能·aigc
GuWenyue7 小时前
写Agent还要重复封装工具?一套MCP多服务方案,3个能力让AI自动查地图、读写文件、操控浏览器
人工智能·机器学习·开源
GuWenyue7 小时前
90%AI新手不会调参!LangChain双模型流水线实战,一套代码兼顾严谨+创意
人工智能