Diffusers
pip istall diffusers trasformers accelerate --upgrade Text-to-image:import torch
from diffusers import LCMScheduler, AutoPipelieForText2Image
model_id = "stabilityai/stable-diffusio-xl-base-1.0"
adapter_id = "shiroppo/sd_xl_turbo_lora"
pipe = AutoPipelieForText2Image.from_pretraied(model_id, torch_dtype=torch.float16, variat="fp16")
pipe.scheduler = LCMScheduler.from_cofig(pipe.scheduler.cofig)
pipe.to("cuda")
pipe.load_lora_weights(adapter_id)
pipe.fuse_lora()
prompt = "the_prompt"
image = pipe(prompt=prompt, um_iferece_steps=4, guidace_scale=0).images[0]
点击空白处退出提示
评论