This commit is contained in:
lyuxiang.lx
2025-06-29 13:21:44 +08:00
parent 46dfe0439b
commit 21a5efd8ae
2 changed files with 2 additions and 2 deletions

View File

@@ -381,7 +381,7 @@ class Qwen2LM(TransformerLM):
self,
batch: dict,
device: torch.device,
) -> Dict[str, Optional[torch.Tensor]]:
) -> Dict[str, Optional[torch.Tensor]]:
text_token = batch['text_token'].to(device)
text_token_len = batch['text_token_len'].to(device)
speech_token = batch['speech_token'].to(device)