refactor: Chat.load_model to Chat.load (2noise#432)

waywayyang · Jun 24, 2024 · 5222976 · 5222976
1 parent e08fd43
commit 5222976
Show file tree

Hide file tree

Showing 9 changed files with 16 additions and 16 deletions.
diff --git a/ChatTTS/core.py b/ChatTTS/core.py
@@ -87,7 +87,7 @@ def download_models(
 
         return download_path
 
-    def load_models(
+    def load(
         self,
         source: Literal['huggingface', 'local', 'custom']='local',
         force_redownload=False,

diff --git a/README.md b/README.md
@@ -108,7 +108,7 @@ from IPython.display import Audio
 import torchaudio
 
 chat = ChatTTS.Chat()
-chat.load_models(compile=False) # Set to True for better performance
+chat.load(compile=False) # Set to True for better performance
 
 texts = ["PUT YOUR TEXT HERE",]
 

diff --git a/docs/cn/README.md b/docs/cn/README.md
@@ -125,7 +125,7 @@ from IPython.display import Audio
 import torchaudio
 
 chat = ChatTTS.Chat()
-chat.load_models(compile=False) # Set to True for better performance
+chat.load(compile=False) # Set to True for better performance
 
 texts = ["PUT YOUR TEXT HERE",]
 

diff --git a/docs/jp/README.md b/docs/jp/README.md
@@ -36,7 +36,7 @@ import ChatTTS
 from IPython.display import Audio
 
 chat = ChatTTS.Chat()
-chat.load_models(compile=False) # より良いパフォーマンスのためにTrueに設定
+chat.load(compile=False) # より良いパフォーマンスのためにTrueに設定
 
 texts = ["ここにテキストを入力してください",]
 

diff --git a/docs/ru/README.md b/docs/ru/README.md
@@ -36,7 +36,7 @@ import ChatTTS
 from IPython.display import Audio
 
 chat = ChatTTS.Chat()
-chat.load_models(compile=False) # Установите значение True для лучшей производительности
+chat.load(compile=False) # Установите значение True для лучшей производительности
 
 texts = ["ВВЕДИТЕ ВАШ ТЕКСТ ЗДЕСЬ",]
 

diff --git a/examples/cmd/run.py b/examples/cmd/run.py
@@ -33,7 +33,7 @@ def main(texts: list[str]):
 
     chat = ChatTTS.Chat(get_logger("ChatTTS"))
     logger.info("Initializing ChatTTS...")
-    if chat.load_models():
+    if chat.load():
         logger.info("Models loaded successfully.")
     else:
         logger.error("Models load failed.")

diff --git a/examples/ipynb/colab.ipynb b/examples/ipynb/colab.ipynb
@@ -123,7 +123,7 @@
       "outputs": [],
       "source": [
         "# use force_redownload=True if the weights have been updated.\n",
-        "chat.load_models(source='huggingface', force_redownload=True)"
+        "chat.load(source='huggingface', force_redownload=True)"
       ]
     },
     {
@@ -143,8 +143,8 @@
       },
       "outputs": [],
       "source": [
-        "chat.load_models()\n",
-        "# chat.load_models(source='local') same as above"
+        "chat.load()\n",
+        "# chat.load(source='local') same as above"
       ]
     },
     {
@@ -165,7 +165,7 @@
       "outputs": [],
       "source": [
         "# write the model path into custom_path\n",
-        "chat.load_models(source='custom', custom_path='YOUR CUSTOM PATH')"
+        "chat.load(source='custom', custom_path='YOUR CUSTOM PATH')"
       ]
     },
     {

diff --git a/examples/ipynb/example.ipynb b/examples/ipynb/example.ipynb
@@ -98,7 +98,7 @@
    "outputs": [],
    "source": [
     "# use force_redownload=True if the weights have been updated.\n",
-    "chat.load_models(source='huggingface', force_redownload=True)"
+    "chat.load(source='huggingface', force_redownload=True)"
    ]
   },
   {
@@ -114,8 +114,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "chat.load_models()\n",
-    "# chat.load_models(source='local') same as above"
+    "chat.load()\n",
+    "# chat.load(source='local') same as above"
    ]
   },
   {
@@ -132,7 +132,7 @@
    "outputs": [],
    "source": [
     "# write the model path into custom_path\n",
-    "chat.load_models(source='custom', custom_path='YOUR CUSTOM PATH')"
+    "chat.load(source='custom', custom_path='YOUR CUSTOM PATH')"
    ]
   },
   {

diff --git a/examples/web/funcs.py b/examples/web/funcs.py
@@ -40,10 +40,10 @@ def on_voice_change(vocie_selection):
 
 def load_chat(cust_path: Optional[str], coef: Optional[str]) -> bool:
     if cust_path == None:
-        ret = chat.load_models(coef=coef, compile=sys.platform != 'win32')
+        ret = chat.load(coef=coef, compile=sys.platform != 'win32')
     else:
         logger.info('local model path: %s', cust_path)
-        ret = chat.load_models('custom', custom_path=cust_path, coef=coef, compile=sys.platform != 'win32')
+        ret = chat.load('custom', custom_path=cust_path, coef=coef, compile=sys.platform != 'win32')
         global custom_path
         custom_path = cust_path
     if ret: