Update config.py

config.py

@@ -15,54 +15,55 @@ def use_fp32_config():
         device = torch.device("cuda:0")  # Assuming you have only one GPU (index 0).
         device_capability = torch.cuda.get_device_capability(device)[0]
         if device_capability >= 7:
-            usefp16 = True
-            for config_file in ["32k.json", "40k.json", "48k.json"]:
-                with open(f"configs/{config_file}", "r") as d:
-                    data = json.load(d)
+        #     usefp16 = True
+        #     for config_file in ["32k.json", "40k.json", "48k.json"]:
+        #         with open(f"configs/{config_file}", "r") as d:
+        #             data = json.load(d)
 
-                if "train" in data and "fp16_run" in data["train"]:
-                    data["train"]["fp16_run"] = True
+        #         if "train" in data and "fp16_run" in data["train"]:
+        #             data["train"]["fp16_run"] = True
 
-                with open(f"configs/{config_file}", "w") as d:
-                    json.dump(data, d, indent=4)
+        #         with open(f"configs/{config_file}", "w") as d:
+        #             json.dump(data, d, indent=4)
 
-                print(f"Set fp16_run to true in {config_file}")
+        #         print(f"Set fp16_run to true in {config_file}")
 
-            with open(
-                "trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
-            ) as f:
-                strr = f.read()
+        #     with open(
+        #         "trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
+        #     ) as f:
+        #         strr = f.read()
 
-            strr = strr.replace("3.0", "3.7")
+        #     strr = strr.replace("3.0", "3.7")
 
-            with open(
-                "trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
-            ) as f:
-                f.write(strr)
-        else:
-            for config_file in ["32k.json", "40k.json", "48k.json"]:
-                with open(f"configs/{config_file}", "r") as f:
-                    data = json.load(f)
+        #     with open(
+        #         "trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
+        #     ) as f:
+        #         f.write(strr)
+        # else:
+        #     for config_file in ["32k.json", "40k.json", "48k.json"]:
+        #         with open(f"configs/{config_file}", "r") as f:
+        #             data = json.load(f)
 
-                if "train" in data and "fp16_run" in data["train"]:
-                    data["train"]["fp16_run"] = False
+        #         if "train" in data and "fp16_run" in data["train"]:
+        #             data["train"]["fp16_run"] = False
 
-                with open(f"configs/{config_file}", "w") as d:
-                    json.dump(data, d, indent=4)
+        #         with open(f"configs/{config_file}", "w") as d:
+        #             json.dump(data, d, indent=4)
 
-                print(f"Set fp16_run to false in {config_file}")
+        #         print(f"Set fp16_run to false in {config_file}")
 
-            with open(
-                "trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
-            ) as f:
-                strr = f.read()
+        #     with open(
+        #         "trainset_preprocess_pipeline_print.py", "r", encoding="utf-8"
+        #     ) as f:
+        #         strr = f.read()
 
-            strr = strr.replace("3.7", "3.0")
+        #     strr = strr.replace("3.7", "3.0")
 
-            with open(
-                "trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
-            ) as f:
-                f.write(strr)
+        #     with open(
+        #         "trainset_preprocess_pipeline_print.py", "w", encoding="utf-8"
+        #     ) as f:
+        #         f.write(strr)
+          pass
     else:
         print(
             "CUDA is not available. Make sure you have an NVIDIA GPU and CUDA installed."