From bd31b5606c413b39d0cad6149c5376805294ec8c Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Tue, 16 Jan 2024 17:04:27 -0500 Subject: [PATCH] Retain suffix (.safetensors, .bin) when renaming a checkpoint file or LoRA - closes #5518 --- invokeai/backend/model_management/model_manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/invokeai/backend/model_management/model_manager.py b/invokeai/backend/model_management/model_manager.py index e9f498a438..362d8d3ff5 100644 --- a/invokeai/backend/model_management/model_manager.py +++ b/invokeai/backend/model_management/model_manager.py @@ -759,7 +759,7 @@ class ModelManager(object): model_type: ModelType, new_name: Optional[str] = None, new_base: Optional[BaseModelType] = None, - ): + ) -> None: """ Rename or rebase a model. """ @@ -781,6 +781,9 @@ class ModelManager(object): # if this is a model file/directory that we manage ourselves, we need to move it if old_path.is_relative_to(self.app_config.models_path): + # keep the suffix! + if old_path.is_file(): + new_name = Path(new_name).with_suffix(old_path.suffix).as_posix() new_path = self.resolve_model_path( Path( BaseModelType(new_base).value,