您好,欢迎来到欧得旅游网。
搜索
您的当前位置:首页Diffusers代码学习:加载UNet和文本编码

Diffusers代码学习:加载UNet和文本编码

来源:欧得旅游网

[~loader.LoraLoaderMixin.load_lora_weights]方法可以将lora权重加载到UNet和文本编码器中。这是加载LoRA的首选方式,因为它可以处理以下情况:

1. LoRA权重对于UNet和文本编码器没有单独的标识符

2. LoRA权重具有UNet和文本编码器的标识符

但是,如果您只需要将LoRA权重加载到UNet中,则可以使用[~loader.UNet2DConditionLoadersMixin.load_attn_procs]方法。

import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
rom diffusers import AutoPipelineForText2Image
import torch

pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",
                                                     torch_dtype=torch.float16).to("cuda")
pipeline.unet.load_attn_procs("jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors")

# use cnmt in the prompt to trigger the LoRA
prompt = "A cute cnmt eating a slice of pizza, stunning color scheme, masterpiece, illustration"
image = pipeline(prompt).images[0]
image.save("load.jpg")

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- ovod.cn 版权所有 湘ICP备2023023988号-4

违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务