From 4c4681c15b4557dd7065cd9e0dc30ecd8562df73 Mon Sep 17 00:00:00 2001 From: Aisuko Date: Sun, 2 Jul 2023 06:05:34 +0000 Subject: [PATCH] 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 78fa848421d8..7684052c313a 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) ```