[Bugfix] Stop reading the removed OmniRequestOutput.request_output accessor (#6172)
Signed-off-by: MrlixiangWE <mrdanaer@gmail.com>
This commit is contained in:
1 parent
fe54ecc388
commit
b95c73742d
12 files changed
+74
-32
No files matched your search
@@ -159,11 +159,10 @@ def _run_once(model: str, yaml_path: str, *, async_mode: bool) -> list[dict]:
|
||||
wav: bytes = b""
|
||||
# Batch size 1: submit a single prompt so scheduling is deterministic.
|
||||
for step in omni.generate([_text_prompt(question)], _sampling_params(), py_generator=False):
|
||||
out = step.request_output
|
||||
if step.final_output_type == "text":
|
||||
text = out.outputs[0].text # exact text, no strip -> truly bit-identical
|
||||
text = step.outputs[0].text # exact text, no strip -> truly bit-identical
|
||||
elif step.final_output_type == "audio":
|
||||
wav = _audio_bytes(out.outputs[0].multimodal_output["audio"])
|
||||
wav = _audio_bytes(step.outputs[0].multimodal_output["audio"])
|
||||
if text is None:
|
||||
raise RuntimeError(f"prompt_id={prompt_id} async={async_mode}: produced no text output")
|
||||
if not wav:
|
||||
|
||||
Reference in new issue
Block a user