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

好了,玩去吧

相关推荐
لا معنى له14 小时前
目标检测的内涵、发展和经典模型--学习笔记
人工智能·笔记·深度学习·学习·目标检测·机器学习
AKAMAI16 小时前
Akamai Cloud客户案例 | CloudMinister借助Akamai实现多云转型
人工智能·云计算
小a杰.17 小时前
Flutter 与 AI 深度集成指南:从基础实现到高级应用
人工智能·flutter
colorknight18 小时前
数据编织-异构数据存储的自动化治理
数据仓库·人工智能·数据治理·数据湖·数据科学·数据编织·自动化治理
Lun3866buzha18 小时前
篮球场景目标检测与定位_YOLO11-RFPN实现详解
人工智能·目标检测·计算机视觉
janefir18 小时前
LangChain框架下DirectoryLoader使用报错zipfile.BadZipFile
人工智能·langchain
齐齐大魔王18 小时前
COCO 数据集
人工智能·机器学习
AI营销实验室20 小时前
原圈科技AI CRM系统赋能销售新未来,行业应用与创新点评
人工智能·科技
爱笑的眼睛1120 小时前
超越MSE与交叉熵:深度解析损失函数的动态本质与高阶设计
java·人工智能·python·ai
tap.AI20 小时前
RAG系列(一) 架构基础与原理
人工智能·架构