Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "111fa990f913a533f5cb235ef8bb9f710539e956"
Unverified Commit cbb1ead6 authored by Apoorva Kulkarni's avatar Apoorva Kulkarni Committed by GitHub
Browse files

docs: Add missing import statement in textual_inversion inference example (#4227) 

docs: Add missing import statement in textual_inversion inference instructions
parent 5470a4fc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/textual_inversion/README.md
View file @ cbb1ead6
...@@ -100,6 +100,7 @@ Once you have trained a model using above command, the inference can be done sim ...@@ -100,6 +100,7 @@ Once you have trained a model using above command, the inference can be done sim
```python ```python
from diffusers import StableDiffusionPipeline from diffusers import StableDiffusionPipeline
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")
......
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