| 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") | |