diff --git a/.github/workflows/test-invoke-conda.yml b/.github/workflows/test-invoke-conda.yml index 69cd7d108da..41f72c5a84d 100644 --- a/.github/workflows/test-invoke-conda.yml +++ b/.github/workflows/test-invoke-conda.yml @@ -115,6 +115,7 @@ jobs: run: | time python scripts/invoke.py \ --no-patchmatch \ + --no-nsfw_checker \ --model ${{ matrix.stable-diffusion-model }} \ --from_file ${{ env.TEST_PROMPTS }} \ --root="${{ env.INVOKEAI_ROOT }}" \ diff --git a/.github/workflows/test-invoke-pip.yml b/.github/workflows/test-invoke-pip.yml index 5d6a447021b..3866a4c32c1 100644 --- a/.github/workflows/test-invoke-pip.yml +++ b/.github/workflows/test-invoke-pip.yml @@ -114,6 +114,7 @@ jobs: run: | time ${{ env.pythonLocation }}/bin/python scripts/invoke.py \ --no-patchmatch \ + --no-nsfw_checker \ --model ${{ matrix.stable-diffusion-model }} \ --from_file ${{ env.TEST_PROMPTS }} \ --root="${{ env.INVOKEAI_ROOT }}" \ diff --git a/ldm/invoke/args.py b/ldm/invoke/args.py index be2eb98a5e0..9fd6d730ef0 100644 --- a/ldm/invoke/args.py +++ b/ldm/invoke/args.py @@ -463,9 +463,12 @@ def _create_arg_parser(self): default='auto', ) model_group.add_argument( + '--nsfw_checker' '--safety_checker', - action='store_true', - help='Check for and blur potentially NSFW images', + action=argparse.BooleanOptionalAction, + dest='safety_checker', + default=False, + help='Check for and blur potentially NSFW images.', ) model_group.add_argument( '--patchmatch',