Add loading text inversion (huggingface#9130)

tolgacangoz · Aug 19, 2024 · 815d882 · 815d882
1 parent ba4348d
commit 815d882
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/textual_inversion/README.md b/examples/textual_inversion/README.md
@@ -109,6 +109,9 @@ import torch
 model_id = "path-to-your-trained-model"
 pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to("cuda")
 
+repo_id_embeds = "path-to-your-learned-embeds"
+pipe.load_textual_inversion(repo_id_embeds)
+
 prompt = "A <cat-toy> backpack"
 
 image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]