Update model.py

model.py

@@ -12,7 +12,6 @@ from diffusers import (
     StableDiffusionControlNetPipeline,
     UniPCMultistepScheduler
 )
-# from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 from cv_utils import resize_image
 from preprocessor import Preprocessor
 from settings import MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES
@@ -59,7 +58,6 @@ class Model:
         controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
             base_model_id, safety_checker=None, controlnet=controlnet, torch_dtype=torch.float16
-            # StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
         )
         pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
         if self.device.type == "cuda":