diff options
author | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 15:46:09 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 15:46:09 +0000 |
commit | 60bd4d52a658838c5ee3f6ddfe8d4db55cf1d764 (patch) | |
tree | b19ce2a07cd9675fc96f9c572116ce0047415ea9 /modules/deepbooru.py | |
parent | 4b0dc206edbad90affe609ac0bf2e9be7e197674 (diff) | |
download | stable-diffusion-webui-gfx803-60bd4d52a658838c5ee3f6ddfe8d4db55cf1d764.tar.gz stable-diffusion-webui-gfx803-60bd4d52a658838c5ee3f6ddfe8d4db55cf1d764.tar.bz2 stable-diffusion-webui-gfx803-60bd4d52a658838c5ee3f6ddfe8d4db55cf1d764.zip |
fix incorrect file extension filter for deepdanbooru models
Diffstat (limited to 'modules/deepbooru.py')
-rw-r--r-- | modules/deepbooru.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index 31ec7e17..dfc83357 100644 --- a/modules/deepbooru.py +++ b/modules/deepbooru.py @@ -21,7 +21,7 @@ class DeepDanbooru: files = modelloader.load_models( model_path=os.path.join(paths.models_path, "torch_deepdanbooru"), model_url='https://github.com/AUTOMATIC1111/TorchDeepDanbooru/releases/download/v1/model-resnet_custom_v3.pt', - ext_filter=".pt", + ext_filter=[".pt"], download_name='model-resnet_custom_v3.pt', ) |