From c5758a89dba9de043779b2fde24d7e37413e3cad Mon Sep 17 00:00:00 2001
From: LINKANG ZHAN <82944614+JunityZhan@users.noreply.github.com>
Date: Sun, 18 Jun 2023 12:39:10 +0800
Subject: [PATCH] Stop extracting features when hubert_base.pt does not exist.
 (#535)

* support detection of pretrained model, support train without pretrained model path in web ui

* support detection of pretrained model, support train without pretrained model path in web ui

* support detection of pretrained model, support train without pretrained model path in web ui

* Stop extracting features when hubert_base.pt is not exist.

* Stop extracting features when hubert_base.pt is not exist.

* Make error more noticeable
---
 extract_feature_print.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/extract_feature_print.py b/extract_feature_print.py
index abf4d25..c6b1c63 100644
--- a/extract_feature_print.py
+++ b/extract_feature_print.py
@@ -63,6 +63,10 @@ def readwave(wav_path, normalize=False):
 
 # HuBERT model
 printt("load model(s) from {}".format(model_path))
+# if hubert model is exist
+if (os.access(model_path, os.F_OK) == False):
+    printt("Error: Extracting is shut down because %s does not exist, you may download it from https://huggingface.co/lj1995/VoiceConversionWebUI/tree/main" % model_path)
+    exit(0)
 models, saved_cfg, task = checkpoint_utils.load_model_ensemble_and_task(
     [model_path],
     suffix="",