from huggingface_hub.repocard import RepoCard | |
from diffusers import DiffusionPipeline | |
import torch | |
lora_model_id = <"lora-sdxl-dreambooth-id"> | |
card = RepoCard.load(lora_model_id) | |
base_model_id = card.data.to_dict()["base_model"] | |
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16) | |
pipe = pipe.to("cuda") | |
pipe.load_lora_weights(lora_model_id) | |
image = pipe("A picture of a sks dog in a bucket", num_inference_steps=25).images[0] | |
image.save("sks_dog.png") | |