add .pt and .pth for available extension

kohya-ss · Sep 24, 2023 · 55886a0 · 55886a0
1 parent 33e90cc
commit 55886a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/networks/resize_lora.py b/networks/resize_lora.py
@@ -283,8 +283,8 @@ 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.save_to is None or not (args.save_to.endswith('.ckpt') or args.save_to.endswith('.pt') or args.save_to.endswith('.pth') or args.save_to.endswith('.safetensors')):
+    raise Exception("The --save_to argument must be specified and must be a .ckpt , .pt, .pth or .safetensors file.")
 
 
   def str_to_dtype(p):