Skip to content

Commit

Permalink
rename infer
Browse files Browse the repository at this point in the history
  • Loading branch information
BeautyyuYanli committed Jun 13, 2024
1 parent 2ec4e85 commit 15c73d6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions src/gpt_sovits/infer/__init__.py
Original file line number Diff line number Diff line change
@@ -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",
]
4 changes: 2 additions & 2 deletions src/gpt_sovits/infer/interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand All @@ -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
Expand Down

0 comments on commit 15c73d6

Please sign in to comment.