update

Author: DN6

scripts/convert_animatediff_motion_lora_to_diffusers.py

@@ -27,15 +27,14 @@ def get_args():
     parser = argparse.ArgumentParser()
     parser.add_argument("--ckpt_path", type=str, required=True)
     parser.add_argument("--output_path", type=str, required=True)
-    parser.add_argument("--use_safetensors", type=str, action="store_true")
 
     return parser.parse_args()
 
 
 if __name__ == "__main__":
     args = get_args()
 
-    if args.use_safetensors:
+    if args.ckpt_path.endswith(".safetensors"):
         state_dict = load_file(args.ckpt_path)
     else:
         state_dict = torch.load(args.ckpt_path, map_location="cpu")

scripts/convert_animatediff_motion_module_to_diffusers.py

@@ -32,15 +32,14 @@ def get_args():
     parser.add_argument("--use_motion_mid_block", action="store_true")
     parser.add_argument("--motion_max_seq_length", type=int, default=32)
     parser.add_argument("--save_fp16", action="store_true")
-    parser.add_argument("--use_safetensors", action="store_true")
 
     return parser.parse_args()
 
 
 if __name__ == "__main__":
     args = get_args()
 
-    if args.use_safetensors:
+    if args.ckpt_path.endswith(".safetensors"):
         state_dict = load_file(args.ckpt_path)
     else:
         state_dict = torch.load(args.ckpt_path, map_location="cpu")