Venevari commited on
Commit
c1ccd2a
1 Parent(s): 1171e3e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -55,8 +55,7 @@ if is_colab:
55
  current_model.path,
56
  torch_dtype=torch.float16,
57
  scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
58
- safety_checker=lambda images, clip_input: (images, False)
59
- )
60
 
61
  else:
62
  eulera = EulerAncestralDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
@@ -175,8 +174,7 @@ def img_to_img(model_path, prompt, neg_prompt, img, strength, guidance, steps, w
175
  current_model_path,
176
  torch_dtype=torch.float16,
177
  scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
178
- safety_checker=lambda images, clip_input: (images, False)
179
- )
180
  else:
181
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
182
  current_model_path,
55
  current_model.path,
56
  torch_dtype=torch.float16,
57
  scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
58
+
 
59
 
60
  else:
61
  eulera = EulerAncestralDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
174
  current_model_path,
175
  torch_dtype=torch.float16,
176
  scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler"),
177
+
 
178
  else:
179
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
180
  current_model_path,