fix anwser
This commit is contained in:
parent
e7f7081bc6
commit
dcbeb65f76
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue