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: