fix: Fix whitelist loading in cmd

Author: Nativu5

configs/config.py

@@ -134,8 +134,8 @@ def use_insecure_load():
         try:
             from fairseq.data.dictionary import Dictionary
 
-            logging.warning("Using insecure weight loading for fairseq dictionary")
             torch.serialization.add_safe_globals([Dictionary])
+            logging.warning("Using insecure weight loading for fairseq dictionary")
         except AttributeError:
             pass
 

infer/modules/train/extract_f0_print.py

@@ -96,6 +96,10 @@ def go(self, paths, f0_method):
     # exp_dir=r"E:\codes\py39\dataset\mi-test"
     # n_p=16
     # f = open("%s/log_extract_f0.log"%exp_dir, "w")
+
+    from configs import Config
+    Config.use_insecure_load()
+    
     printt(" ".join(sys.argv))
     featureInput = FeatureInput(is_half, device)
     paths = []

infer/modules/train/extract_feature_print.py

@@ -23,11 +23,15 @@
     os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
     version = sys.argv[6]
     is_half = sys.argv[7].lower() == "true"
+
 import fairseq
 import numpy as np
 import torch
 import torch.nn.functional as F
 
+from configs import Config
+Config.use_insecure_load()
+
 if "privateuseone" not in device:
     device = "cpu"
     if torch.cuda.is_available():

infer/modules/train/preprocess.py

@@ -142,4 +142,6 @@ def preprocess_trainset(inp_root, sr, n_p, exp_dir, per):
 
 
 if __name__ == "__main__":
+    from configs import Config
+    Config.use_insecure_load()
     preprocess_trainset(inp_root, sr, n_p, exp_dir, per)