Thafx commited on
Commit
4db70d1
1 Parent(s): 9178799

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14
  scheduler=scheduler,
15
- pipe.safety_checker=lambda images, clip_input: (images, False))
16
 
17
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
18
  model_id,
 
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14
  scheduler=scheduler,
15
+ pipe.safety_checker=lambda )
16
 
17
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
18
  model_id,