Skip to content

Commit b298484

Browse files
authored
fix/doc: no import torch issue (#3923)
Ffix/doc: no import torch issue Signed-off-by: GitHub <[email protected]>
1 parent f911287 commit b298484

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/source/en/stable_diffusion.mdx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ pipeline = pipeline.to("cuda")
5252
To make sure you can use the same image and improve on it, use a [`Generator`](https://pytorch.org/docs/stable/generated/torch.Generator.html) and set a seed for [reproducibility](./using-diffusers/reproducibility):
5353

5454
```python
55+
import torch
56+
5557
generator = torch.Generator("cuda").manual_seed(0)
5658
```
5759

0 commit comments

Comments
 (0)