diff --git a/lama_cleaner/model/sdxl.py b/lama_cleaner/model/sdxl.py index d55c591..05fd9cf 100644 --- a/lama_cleaner/model/sdxl.py +++ b/lama_cleaner/model/sdxl.py @@ -22,15 +22,6 @@ class SDXL(DiffusionInpaintModel): model_kwargs = { "local_files_only": kwargs.get("local_files_only", kwargs["sd_run_local"]) } - if kwargs["disable_nsfw"] or kwargs.get("cpu_offload", False): - logger.info("Disable Stable Diffusion Model NSFW checker") - model_kwargs.update( - dict( - safety_checker=None, - feature_extractor=None, - requires_safety_checker=False, - ) - ) use_gpu = device == torch.device("cuda") and torch.cuda.is_available() torch_dtype = torch.float16 if use_gpu and fp16 else torch.float32