This commit is contained in:
游雁 2024-07-11 10:49:29 +08:00
parent 1e9e3f864e
commit 6ab5a81ef8

View File

@ -1939,6 +1939,9 @@ class LLMASR5(nn.Module):
if role == "system":
system.append(content)
elif role == "user":
if "audio" in item:
audio = item["audio"]
content = [content, audio]
user.append(content)
elif role == "assistant":
assistant.append(content)