chore(format): run black on dev (#719)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2noise · Aug 25, 2024 · 9f9abec · 9f9abec
1 parent 2d8fde1
commit 9f9abec
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/ChatTTS/config/config.py b/ChatTTS/config/config.py
@@ -62,6 +62,7 @@ class GPT:
     num_text_tokens: int = 21178
     num_vq: int = 4
 
+
 @dataclass(repr=False, eq=False)
 class Embed:
     hidden_size: int = 768

diff --git a/ChatTTS/model/gpt.py b/ChatTTS/model/gpt.py
@@ -56,7 +56,9 @@ def __init__(
         self.head_text = embed.head_text.__call__
         self.head_code = [hc.__call__ for hc in embed.head_code]
 
-    def from_pretrained(self, gpt_folder: str, embed_file_path: str, experimental=False):
+    def from_pretrained(
+        self, gpt_folder: str, embed_file_path: str, experimental=False
+    ):
         if self.is_vllm and platform.system().lower() == "linux":
 
             from .velocity import LLM
@@ -70,7 +72,9 @@ def from_pretrained(self, gpt_folder: str, embed_file_path: str, experimental=Fa
             self.logger.info("vLLM model loaded")
             return
 
-        self.gpt: LlamaModel = LlamaModel.from_pretrained(gpt_folder).to(self.device_gpt)
+        self.gpt: LlamaModel = LlamaModel.from_pretrained(gpt_folder).to(
+            self.device_gpt
+        )
         del self.gpt.embed_tokens
 
         if (

diff --git a/ChatTTS/model/velocity/sampler.py b/ChatTTS/model/velocity/sampler.py
@@ -4,6 +4,7 @@
 
 from ..embed import Embed
 
+
 class Sampler:
     def __init__(self, post_model: Embed, num_audio_tokens: int, num_vq: int):
         self.post_model = post_model