AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
docs: Add missing import statement in textual_inversion inference instructions
This commit is contained in:
Apoorva Kulkarni
2023-07-24 11:07:53 -07:00
committed by GitHub
parent 5470a4fce3
commit cbb1ead60b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/textual_inversion/README.md
View File

@@ -100,6 +100,7 @@ Once you have trained a model using above command, the inference can be done sim
```python
from diffusers import StableDiffusionPipeline
import torch
model_id = "path-to-your-trained-model"
pipe = StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16).to("cuda")