disable NSFW checker loading during the CI tests (#1641)

* disable NSFW checker loading during the CI tests

The NSFW filter apparently causes invoke.py to crash during CI testing,
possibly due to out of memory errors. This workaround disables NSFW
model loading.

* doc change

* fix formatting errors in yml files
This commit is contained in:
Lincoln Stein 2022-11-30 14:11:51 -05:00 committed by GitHub
parent 5e81f51f6a
commit 22133392b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 2 deletions

View File

@ -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 }}" \

View File

@ -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 }}" \

View File

@ -463,9 +463,12 @@ class Args(object):
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',