fix anwser

This commit is contained in:
itqop 2025-03-03 04:56:03 +03:00
parent e7f7081bc6
commit dcbeb65f76
1 changed files with 1 additions and 1 deletions

View File

@ -46,5 +46,5 @@ class InferenceService:
early_stopping=True, early_stopping=True,
no_repeat_ngram_size=2 no_repeat_ngram_size=2
) )
responses = [self.tokenizer.decode(out, skip_special_tokens=True) for out in outputs] responses = [self.tokenizer.decode(out, skip_special_tokens=True).replace("Ожидаемый ответ:", "").strip() for out in outputs]
return responses return responses