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
...@@ -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