Update process_ckpt.py

2023-04-09 15:24:46 +00:00 · 2023-04-09 15:24:46 +00:00 · 27c7e3f6a3
parent 541837eb0f
commit 27c7e3f6a3
1 changed files with 1 additions and 1 deletions
--- 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: