-
Notifications
You must be signed in to change notification settings - Fork 6.5k
Description
Hi all,
I have pushed the weights of my Dreambooth model to Huggingface (I used LoRa model). I confirm that pytorch_lora_weights.safetensors is indeed on Huggingface. When I want to get inference from the model, I run the following (as per the Readme file):
`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")`
But I keep getting this error:
OSError: We couldn't connect to 'https://huggingface.co' to load this model, couldn't find it in the cached files and it looks like "path" is not the path to a directory containing a file named pytorch_lora_weights.safetensors or
Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/diffusers/installation#offline-mode'.
Can you please let me know how to resolve this issue?