Skip to content

Commit 52e8fdb

Browse files
Update README.md
1 parent ed6c61c commit 52e8fdb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/community/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def download_image(url):
133133
response = requests.get(url)
134134
return PIL.Image.open(BytesIO(response.content)).convert("RGB")
135135

136-
pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", custom_pipeline="stable_diffusion_mega", dtype=torch.float16, revision="fp16")
136+
pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float16, revision="fp16")
137137
pipe.to("cuda")
138138
pipe.enable_attention_slicing()
139139

0 commit comments

Comments
 (0)