fix crash when text mask applied to img2img (#2526)

This PR fixes the crash reported at https://discord.com/channels/1020123559063990373/1031668022294884392/1071782238137630800

It also quiets-down the "NSFW is disabled" nag during img2img generation.
This commit is contained in:
Lincoln Stein
2023-02-05 10:26:40 -05:00
committed by GitHub
parent 82ae2769c8
commit 4895fe8395
2 changed files with 4 additions and 1 deletions

View File

@@ -71,6 +71,8 @@ def main():
# when the frozen CLIP tokenizer is imported
import transformers
transformers.logging.set_verbosity_error()
import diffusers
diffusers.logging.set_verbosity_error()
# Loading Face Restoration and ESRGAN Modules
gfpgan,codeformer,esrgan = load_face_restoration(opt)