diff --git a/examples/community/README.md b/examples/community/README.md index dc35d36a95..6e0faf5746 100644 --- a/examples/community/README.md +++ b/examples/community/README.md @@ -370,7 +370,7 @@ def dummy(images, **kwargs): pipe.safety_checker = dummy images = [] -generator = th.Generator("cuda").manual_seed(0) +generator = torch.Generator("cuda").manual_seed(0) seed = 0 prompt = "a forest | a camel" @@ -652,4 +652,4 @@ prompt = "a cup" # the masked out region will be replaced with this with autocast("cuda"): image = pipe(image=image, text=text, prompt=prompt).images[0] -``` \ No newline at end of file +```