nateraw commited on
Commit
17f3690
1 Parent(s): 5b675db

disable safety checker

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ pipeline = StableDiffusionWalkPipeline.from_pretrained(
13
  revision="fp16",
14
  scheduler=LMSDiscreteScheduler(
15
  beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear"
16
- )
 
17
  ).to("cuda")
18
  pipeline.enable_attention_slicing()
19
 
 
13
  revision="fp16",
14
  scheduler=LMSDiscreteScheduler(
15
  beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear"
16
+ ),
17
+ safety_checker=None
18
  ).to("cuda")
19
  pipeline.enable_attention_slicing()
20