mirror of
https://github.com/2noise/ChatTTS.git
synced 2026-08-31 01:12:28 +08:00
chore(format): run black on dev (#629)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
committed by
GitHub
parent
2bfb0977d2
commit
06b823be8b
+3
-1
@@ -635,7 +635,9 @@ class Chat:
|
||||
input_ids_list = [i.tolist() for i in input_ids]
|
||||
del input_ids
|
||||
|
||||
result = gpt.llm.generate(None, sample_params, input_ids_list, params.show_tqdm)
|
||||
result = gpt.llm.generate(
|
||||
None, sample_params, input_ids_list, params.show_tqdm
|
||||
)
|
||||
token_ids = []
|
||||
hidden_states = []
|
||||
for i in result:
|
||||
|
||||
@@ -571,7 +571,7 @@ class ModelRunner:
|
||||
logprob_i = logprob[i].cpu().tolist()
|
||||
tmp_hidden_states = hidden_states[i].cpu()
|
||||
if input_tokens[i].shape[-2] != 1:
|
||||
tmp_hidden_states = tmp_hidden_states[-1:,:]
|
||||
tmp_hidden_states = tmp_hidden_states[-1:, :]
|
||||
result = SequenceGroupOutput(
|
||||
samples=[
|
||||
SequenceOutput(
|
||||
|
||||
Reference in New Issue
Block a user