espnet2.sds.end_to_end.mini_omni.litgpt.generate.base.generate_TA_BATCH

source

espnet2.sds.end_to_end.mini_omni.litgpt.generate.base.generate_TA_BATCH(model: GPT, audio_features: Tensor, input_ids: list, leng, task, max_returned_tokens: int = 1000, , temperature: float = 1.0, top_k: int | None = None, top_p: float = 1.0, eos_id_a: int | None = None, eos_id_t: int | None = None, pad_id_t: int | None = None, shift: int | None = None, include_prompt: bool = True, generate_text=False) → Tensor