fix incorrect file extension filter for deepdanbooru models

3 years ago · 60bd4d52a6
parent 4b0dc206ed
commit 60bd4d52a6
1 changed files with 1 additions and 1 deletions
--- 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',
        )