fix incorrect file extension filter for deepdanbooru models

This commit is contained in:
AUTOMATIC 2022-12-03 18:46:09 +03:00
parent 4b0dc206ed
commit 60bd4d52a6

View File

@ -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',
)