From b298484fd0303ba48f1fdceff3100d7068f62b79 Mon Sep 17 00:00:00 2001 From: Aisuko Date: Mon, 3 Jul 2023 20:28:42 +1000 Subject: [PATCH] fix/doc: no import torch issue (#3923) Ffix/doc: no import torch issue Signed-off-by: GitHub --- docs/source/en/stable_diffusion.mdx | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/source/en/stable_diffusion.mdx b/docs/source/en/stable_diffusion.mdx index 78fa848421..7684052c31 100644 --- a/docs/source/en/stable_diffusion.mdx +++ b/docs/source/en/stable_diffusion.mdx @@ -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): ```python +import torch + generator = torch.Generator("cuda").manual_seed(0) ```