|
|
import os |
|
|
|
|
|
|
|
|
os.environ["SAFETY_CHECKER"] = "False" |
|
|
os.environ["DISABLE_SAFETY_CHECKER"] = "1" |
|
|
|
|
|
try: |
|
|
|
|
|
from diffusers.pipelines.stable_diffusion import safety_checker |
|
|
def dummy_safety_checker(self, clip_input, images): |
|
|
return images, [False] * len(images) |
|
|
safety_checker.StableDiffusionSafetyChecker.forward = dummy_safety_checker |
|
|
print("NSFW Filter has been disabled successfully!") |
|
|
except Exception as e: |
|
|
print(f"Could not disable safety checker: {e}") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
from frontend.webui.hf_demo import start_demo_text_to_image |
|
|
|
|
|
print("Starting HF demo text to image") |
|
|
|
|
|
start_demo_text_to_image(False) |