Hugging Face's logo

Spaces:
AP123
/
IllusionDiffusion
Running on Zero

App Files Community
1053

temporarily disable safety checker to fix breaking error

#1074
by linoyts HF Staff - opened
base: refs/heads/main
←
from: refs/pr/1074
Discussion Files changed
+2
-1
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=tor
32
  controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16)
33
 
34
  # Initialize the safety checker conditionally
35
- SAFETY_CHECKER_ENABLED = os.environ.get("SAFETY_CHECKER", "0") == "1"
 
36
  safety_checker = None
37
  feature_extractor = None
38
  if SAFETY_CHECKER_ENABLED:
 
32
  controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16)
33
 
34
  # Initialize the safety checker conditionally
35
+ # SAFETY_CHECKER_ENABLED = os.environ.get("SAFETY_CHECKER", "0") == "1"
36
+ SAFETY_CHECKER_ENABLED = False
37
  safety_checker = None
38
  feature_extractor = None
39
  if SAFETY_CHECKER_ENABLED: