How to determine which model textual inversion is using

🧨 Diffusers
1

I’m using the example code in textual inversion to generate images:

from diffusers import StableDiffusionPipeline

model_id = "path-to-your-trained-model"
pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to("cuda")

prompt = "A <cat-toy> backpack"

image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]

image.save("cat-backpack.png")

How do I display /determine which model (1.4/1.5/etc) that StableDiffusionPipeline is using. When I add the model here it is just the generated model.

Thanks for your time!

2

Is there a better place to ask this question?