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

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

ChatTTS/model/cuda/te_llama.py

@@ -112,7 +112,9 @@ def __new__(cls, config: LlamaConfig):
 
     @classmethod
     def from_state_dict(
-        cls, state_dict: Dict[str, torch.Tensor], config: LlamaConfig,
+        cls,
+        state_dict: Dict[str, torch.Tensor],
+        config: LlamaConfig,
     ):
         """
         Custom method adapted from `from_pretrained` method in HuggingFace

ChatTTS/model/gpt.py

@@ -81,12 +81,14 @@ def __init__(
                 for _ in range(self.num_vq)
             ],
         )
-    
+
     def from_pretrained(self, file_path: str):
 
         self.load_state_dict(torch.load(file_path, weights_only=True, mmap=True))
 
-        if "cuda" in str(self.device_gpt) and platform.system().lower() == "linux": # is TELlamaModel
+        if (
+            "cuda" in str(self.device_gpt) and platform.system().lower() == "linux"
+        ):  # is TELlamaModel
             try:
                 from .cuda import TELlamaModel