diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 18:46:09 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-12-03 18:46:09 +0300 |
| commit | 60bd4d52a658838c5ee3f6ddfe8d4db55cf1d764 (patch) | |
| tree | b19ce2a07cd9675fc96f9c572116ce0047415ea9 | |
| parent | 4b0dc206edbad90affe609ac0bf2e9be7e197674 (diff) | |
fix incorrect file extension filter for deepdanbooru models
| -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', ) |