alexandergar0 commited on
Commit
4e3f093
1 Parent(s): 6a603f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,8 @@ from diffusers import StableDiffusionPipeline
6
  #model_id = "hakurei/waifu-diffusion"
7
  pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
 
 
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2
11
 
 
6
  #model_id = "hakurei/waifu-diffusion"
7
  pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
9
+ def dummy_checker(images, **kwargs): return images, False
10
+ pipe.safety_checker = dummy_checker
11
  #torch.backends.cudnn.benchmark = True
12
  num_samples = 2
13