Skip to content

Commit f499721

Browse files
Update examples/community/README.md
Co-authored-by: Suraj Patil <[email protected]>
1 parent bc0cd0c commit f499721

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/community/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,7 @@ device = torch.device('cpu' if not has_cuda else 'cuda')
388388

389389
pipe = DiffusionPipeline.from_pretrained(
390390
"CompVis/stable-diffusion-v1-4",
391+
safety_checker=None,
391392
use_auth_token=True,
392393
custom_pipeline="imagic_stable_diffusion",
393394
scheduler = DDIMScheduler.from_pretrained("CompVis/stable-diffusion-v1-4", subfolder="scheduler")

0 commit comments

Comments
 (0)