diff --git a/src/gpt_sovits/infer/__init__.py b/src/gpt_sovits/infer/__init__.py index 09f5e954..b8489e86 100644 --- a/src/gpt_sovits/infer/__init__.py +++ b/src/gpt_sovits/infer/__init__.py @@ -1,9 +1,9 @@ from gpt_sovits.infer.inference import GPTSoVITSInference from gpt_sovits.infer.inference_pool import GPTSoVITSInferencePool -from gpt_sovits.infer.interface import GPTSoVITSInferenceSimple +from gpt_sovits.infer.interface import GPTSoVITSInterfaceSimple __all__ = [ "GPTSoVITSInference", "GPTSoVITSInferencePool", - "GPTSoVITSInferenceSimple", + "GPTSoVITSInterfaceSimple", ] diff --git a/src/gpt_sovits/infer/interface.py b/src/gpt_sovits/infer/interface.py index ce96f301..64a80749 100644 --- a/src/gpt_sovits/infer/interface.py +++ b/src/gpt_sovits/infer/interface.py @@ -51,7 +51,7 @@ def read_config_data(config_data_base: str) -> ConfigData: PromptType = Union[str, Tuple[str, int, "np.ndarray"]] -class GPTSoVITSInferenceSimple: +class GPTSoVITSInterfaceSimple: config_data: ConfigData working_model: Optional[str] working_prompt: Optional[str] @@ -63,7 +63,7 @@ def __init__( self, config_data_base: str, device: Optional[str] = None, - is_half: Optional[bool] = False, + is_half: Optional[bool] = True, ): self.config_data = read_config_data(config_data_base) self.working_model = None