"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "4c54519e1a640f393ff790a72be38284d4253b45"
Unverified Commit b298484f authored by Aisuko's avatar Aisuko Committed by GitHub
Browse files

fix/doc: no import torch issue (#3923)



Ffix/doc: no import torch issue
Signed-off-by: default avatarGitHub <noreply@github.com>
parent f911287c
...@@ -52,6 +52,8 @@ pipeline = pipeline.to("cuda") ...@@ -52,6 +52,8 @@ pipeline = pipeline.to("cuda")
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): 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):
```python ```python
import torch
generator = torch.Generator("cuda").manual_seed(0) generator = torch.Generator("cuda").manual_seed(0)
``` ```
......
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