Refactor: rename Model / ModelHolder to TTSModel / TTSModelHolder for clarification and add comments to each method
This commit is contained in:
@@ -12,7 +12,7 @@ from tqdm import tqdm
|
||||
|
||||
from config import config
|
||||
from style_bert_vits2.logging import logger
|
||||
from style_bert_vits2.tts_model import Model
|
||||
from style_bert_vits2.tts_model import TTSModel
|
||||
|
||||
warnings.filterwarnings("ignore")
|
||||
|
||||
@@ -54,7 +54,7 @@ safetensors_files = model_path.glob("*.safetensors")
|
||||
|
||||
|
||||
def get_model(model_file: Path):
|
||||
return Model(
|
||||
return TTSModel(
|
||||
model_path=str(model_file),
|
||||
config_path=str(model_file.parent / "config.json"),
|
||||
style_vec_path=str(model_file.parent / "style_vectors.npy"),
|
||||
|
||||
Reference in New Issue
Block a user