Update convert.py

convert.py

@@ -9,7 +9,7 @@ from typing import Optional
 
 import torch
 from io import BytesIO
-
+from safetensors.torch import save_file
 from huggingface_hub import CommitInfo, Discussion, HfApi, hf_hub_download
 from huggingface_hub.file_download import repo_folder_name
 import spaces