From 27c7e3f6a3f25bf0925282dfaabbfeea44094dc8 Mon Sep 17 00:00:00 2001
From: liujing04 <129054828+liujing04@users.noreply.github.com>
Date: Sun, 9 Apr 2023 15:24:46 +0000
Subject: [PATCH] Update process_ckpt.py

---
 train/process_ckpt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train/process_ckpt.py b/train/process_ckpt.py
index 849afe5..b0b8f66 100644
--- a/train/process_ckpt.py
+++ b/train/process_ckpt.py
@@ -41,7 +41,7 @@ def extract_small_model(path,name,sr,if_f0,info):
         if(info==""):info="Extracted model."
         opt["info"] = info
         opt["sr"] = sr
-        opt["f0"] =if_f0
+        opt["f0"] =int(if_f0)
         torch.save(opt, "weights/%s.pth"%name)
         return "Success."
     except: