From e33c007cd0be3d3fa341ca8e0dcc132184bfd7d7 Mon Sep 17 00:00:00 2001 From: jvkap <83289567+jvkap@users.noreply.github.com> Date: Mon, 11 Sep 2023 11:29:06 -0300 Subject: [PATCH 1/2] Update resize_lora.py --- networks/resize_lora.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/networks/resize_lora.py b/networks/resize_lora.py index 7b740634..9f0207b9 100644 --- a/networks/resize_lora.py +++ b/networks/resize_lora.py @@ -283,7 +283,12 @@ def resize_lora_model(lora_sd, new_rank, save_dtype, device, dynamic_method, dyn def resize(args): - + if args.save_to is None or not (args.save_to.endswith('.ckpt') or args.save_to.endswith('.safetensors')): + raise Exception("The --save_to argument must be specified and must be a .ckpt or .safetensors file.") + + if args.model is None or not (args.model.endswith('.ckpt') or args.model.endswith('.safetensors')): + raise Exception("The --model argument must be specified and must be a .ckpt or .safetensors file.") + def str_to_dtype(p): if p == 'float': return torch.float From a0e05fa291039f7a81a34cdf565d20219e8c77c6 Mon Sep 17 00:00:00 2001 From: jvkap <83289567+jvkap@users.noreply.github.com> Date: Mon, 11 Sep 2023 11:41:33 -0300 Subject: [PATCH 2/2] Update resize_lora.py --- networks/resize_lora.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/networks/resize_lora.py b/networks/resize_lora.py index 9f0207b9..41585f79 100644 --- a/networks/resize_lora.py +++ b/networks/resize_lora.py @@ -285,9 +285,7 @@ def resize_lora_model(lora_sd, new_rank, save_dtype, device, dynamic_method, dyn def resize(args): if args.save_to is None or not (args.save_to.endswith('.ckpt') or args.save_to.endswith('.safetensors')): raise Exception("The --save_to argument must be specified and must be a .ckpt or .safetensors file.") - - if args.model is None or not (args.model.endswith('.ckpt') or args.model.endswith('.safetensors')): - raise Exception("The --model argument must be specified and must be a .ckpt or .safetensors file.") + def str_to_dtype(p): if p == 'float':