diff --git a/ChatTTS/model/gpt.py b/ChatTTS/model/gpt.py
index d66106231..4ae236c60 100644
--- a/ChatTTS/model/gpt.py
+++ b/ChatTTS/model/gpt.py
@@ -468,7 +468,7 @@ def generate(
                         )
                         for num_vq_iter in range(self.num_vq):
                             x: torch.Tensor = self.head_code[num_vq_iter](hidden_states)
-                            logits[..., num_vq_iter ] = x
+                            logits[..., num_vq_iter] = x
                             del x
 
                 # logits = logits[:, -1].float()
@@ -524,11 +524,16 @@ def generate(
                         ],
                         1,
                     )
-                
+
                 if i == 0 and finish.any():
                     self.logger.warn(
                         "unexpected end at index %s",
-                        str([unexpected_idx.item() for unexpected_idx in finish.nonzero()]),
+                        str(
+                            [
+                                unexpected_idx.item()
+                                for unexpected_idx in finish.nonzero()
+                            ]
+                        ),
                     )
                     if ensure_non_empty:
                         if show_tqdm:
diff --git a/tests/#511.py b/tests/#511.py
index 01b7829da..cea9127ef 100644
--- a/tests/#511.py
+++ b/tests/#511.py
@@ -15,15 +15,24 @@
 logger = get_logger("Test #511", lv=logging.WARN)
 
 chat = ChatTTS.Chat(logger)
-chat.load(compile=False, source='huggingface') # Set to True for better performance
-
-texts = ['的 话 语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 ， 老 大 爷 觉 得 车 夫 的 想 法 很 有 道 理 [uv_break]', '的 话 评 分 只 是 衡 量 音 色 的 稳 定 性 ， 不 代 表 音 色 的 好 坏 ， 可 以 根 据 自 己 的 需 求 选 择 [uv_break] 合 适 的 音 色', '然 后 举 个 简 单 的 例 子 ， 如 果 一 个 [uv_break] 沙 哑 且 结 巴 的 音 色 一 直 很 稳 定 ， 那 么 它 的 评 分 就 会 很 高 。', '语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 [uv_break] 占 位 。 我 使 用 seed id 去 生 成 音 频 ， 但 是 生 成 的 音 频 不 稳 定', '在d id 只 是 一 个 参 考 id [uv_break] 不 同 的 环 境 下 音 色 不 一 定 一 致 。 还 是 推 荐 使 用 。 pt 文 件 载 入 音 色', '的 话 语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 。 音 色 标 的 男 女 [uv_break] 准 确 吗', '， 当 前 第 一 批 测 试 的 音 色 有 两 千 条 [uv_break] ， 根 据 声 纹 相 似 性 简 单 打 标 ， 准 确 度 不 高 ， 特 别 是 特 征 一 项', '语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 。 仅 供 参 考 。 如 果 大 家 有 更 好 的 标 注 方 法 ， 欢 迎 pr [uv_break] 。']
+chat.load(compile=False, source="huggingface")  # Set to True for better performance
+
+texts = [
+    "的 话 语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 ， 老 大 爷 觉 得 车 夫 的 想 法 很 有 道 理 [uv_break]",
+    "的 话 评 分 只 是 衡 量 音 色 的 稳 定 性 ， 不 代 表 音 色 的 好 坏 ， 可 以 根 据 自 己 的 需 求 选 择 [uv_break] 合 适 的 音 色",
+    "然 后 举 个 简 单 的 例 子 ， 如 果 一 个 [uv_break] 沙 哑 且 结 巴 的 音 色 一 直 很 稳 定 ， 那 么 它 的 评 分 就 会 很 高 。",
+    "语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 [uv_break] 占 位 。 我 使 用 seed id 去 生 成 音 频 ， 但 是 生 成 的 音 频 不 稳 定",
+    "在d id 只 是 一 个 参 考 id [uv_break] 不 同 的 环 境 下 音 色 不 一 定 一 致 。 还 是 推 荐 使 用 。 pt 文 件 载 入 音 色",
+    "的 话 语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 。 音 色 标 的 男 女 [uv_break] 准 确 吗",
+    "， 当 前 第 一 批 测 试 的 音 色 有 两 千 条 [uv_break] ， 根 据 声 纹 相 似 性 简 单 打 标 ， 准 确 度 不 高 ， 特 别 是 特 征 一 项",
+    "语 音 太 短 了 会 造 成 生 成 音 频 错 误 ， 这 是 占 位 占 位 。 仅 供 参 考 。 如 果 大 家 有 更 好 的 标 注 方 法 ， 欢 迎 pr [uv_break] 。",
+]
 
 params_infer_code = ChatTTS.Chat.InferCodeParams(
-    spk_emb = chat.sample_random_speaker(),
-    temperature = .3,
-    top_P = 0.005,
-    top_K = 1,
+    spk_emb=chat.sample_random_speaker(),
+    temperature=0.3,
+    top_P=0.005,
+    top_K=1,
     show_tqdm=False,
 )
 
@@ -42,4 +51,5 @@
 
 if fail:
     import sys
+
     sys.exit(1)