From b61c83e8369b34cf9546f98cdd0372d053b96099 Mon Sep 17 00:00:00 2001 From: Sergey Borisov Date: Mon, 17 Jul 2023 00:32:17 +0300 Subject: [PATCH] Allow bin extension to detect diffusers-ti provided as file --- invokeai/backend/model_management/models/textual_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_management/models/textual_inversion.py b/invokeai/backend/model_management/models/textual_inversion.py index 9cd62bb417..eea0e85245 100644 --- a/invokeai/backend/model_management/models/textual_inversion.py +++ b/invokeai/backend/model_management/models/textual_inversion.py @@ -68,7 +68,7 @@ class TextualInversionModel(ModelBase): return None # diffusers-ti if os.path.isfile(path): - if any([path.endswith(f".{ext}") for ext in ["safetensors", "ckpt", "pt"]]): + if any([path.endswith(f".{ext}") for ext in ["safetensors", "ckpt", "pt", "bin"]]): return None raise InvalidModelException(f"Not a valid model: {path}")