From 01ea5451fad0878fa2d4ba71101680b9c44432b5 Mon Sep 17 00:00:00 2001
From: Wenlong Wu <wlwu1993@gmail.com>
Date: Thu, 8 Aug 2024 09:04:40 +0800
Subject: [PATCH] Add loading text inversion

---
 examples/textual_inversion/README.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/examples/textual_inversion/README.md b/examples/textual_inversion/README.md
index 9e3a622943a1..3287f698b870 100644
--- 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]