Skip to content

GitLab

Unverified Commit 815d8822 authored by Wenlong Wu's avatar Wenlong Wu Committed by GitHub
Browse files

Add loading text inversion (#9130)

parent ba4348d9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
examples/textual_inversion/README.md
View file @ 815d8822
...@@ -109,6 +109,9 @@ import torch ...@@ -109,6 +109,9 @@ import torch
model_id = "path-to-your-trained-model" model_id = "path-to-your-trained-model"
pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to("cuda") 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" prompt = "A <cat-toy> backpack"
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0] image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment