This commit is contained in:
itqop 2025-03-03 04:59:00 +03:00
parent 6dc88090fb
commit ed203a9fca
1 changed files with 1 additions and 1 deletions

View File

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