Skip to content

Commit 69351de

Browse files
EYH0602Copilot
andcommitted
Update src/tfbench/lm/_hf.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 5ef72f9 commit 69351de

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

src/tfbench/lm/_hf.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,5 @@ def _gen(self, prompt: str) -> LMAnswer:
4949
output_ids = generated_ids[0][len(model_inputs.input_ids[0]) :].tolist()
5050
output = self.tokenizer.decode(output_ids, skip_special_tokens=True).strip("\n")
5151

52-
if output is None:
53-
raise NoneResponseError(self.model_name)
54-
5552
content, thinking_content = extract_thinking_content(output)
5653
return LMAnswer(answer=content, reasoning_steps=thinking_content)

0 commit comments

Comments
 (0)