Izac commited on
Commit
d0dd877
·
verified ·
1 Parent(s): 1e92513

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,10 +72,10 @@ class LazyLoadPipeline:
72
  @timer_func
73
  def setup_pipeline(self):
74
  print("Setting up the pipeline...")
75
- controlnet = ControlNetModel.from_single_file(
76
  "models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16
77
  )
78
- safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
79
  model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
80
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
81
  model_path,
 
72
  @timer_func
73
  def setup_pipeline(self):
74
  print("Setting up the pipeline...")
75
+ controlnet = ControlNetModel(
76
  "models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16
77
  )
78
+ safety_checker = None
79
  model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
80
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
81
  model_path,