Fast_Stable_diffusion_CPU

Running

charliebaby2023 commited on Apr 19

Commit

153faea

verified ·

1 Parent(s): 5653fac

Update scripts/main.py

Files changed (1) hide show

scripts/main.py CHANGED Viewed

@@ -161,7 +161,7 @@ def generate_i2i(
         tokenizer = pipe.tokenizer,
         unet = pipe.unet,
         scheduler = None, #scheduler,
-        safety_checker = pipe.safety_checker,
         feature_extractor = pipe.feature_extractor,
         requires_safety_checker = False,
     )
@@ -283,7 +283,7 @@ def generate_v2v(
         tokenizer = pipe.tokenizer,
         unet = pipe.unet,
         scheduler = None,
-        safety_checker = pipe.safety_checker,
         feature_extractor = pipe.feature_extractor,
         requires_safety_checker = False,
     )

         tokenizer = pipe.tokenizer,
         unet = pipe.unet,
         scheduler = None, #scheduler,
+        safety_checker=None,  # Disable NSFW filter
         feature_extractor = pipe.feature_extractor,
         requires_safety_checker = False,
     )
         tokenizer = pipe.tokenizer,
         unet = pipe.unet,
         scheduler = None,
+        safety_checker=None,  # Disable NSFW filter
         feature_extractor = pipe.feature_extractor,
         requires_safety_checker = False,
     )