fix

CNChTu · Jan 31, 2024 · 2d9964b · 2d9964b
1 parent 03fe7f5
commit 2d9964b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tools/combine_tools.py b/tools/combine_tools.py
@@ -67,7 +67,8 @@ def save_combo_model(self, save_path, save_name):
         if self.is_combo_trained_model:
             save_dict = {
                 "diff_model": self.diff_model,
-                "diff_config_dict": self.diff_config_dict
+                "diff_config_dict": self.diff_config_dict,
+                "_is_comb_diff_model": True
             }
         else:
             save_dict = {
@@ -77,7 +78,7 @@ def save_combo_model(self, save_path, save_name):
                 "naive_config_dict": self.naive_config_dict
             }
         if self.vocoder is not None:
-            print(self.vocoder['model'])
+            print(" [INFO] Saving vocoder model")
             save_dict["vocoder"] = self.vocoder
             save_dict["vocoder_type"] = self.vocoder_type
         torch.save(save_dict, out_path)