fix hf conversational task bug

This commit is contained in:
Krrish Dholakia 2023-09-18 11:46:14 -07:00
parent 154aa83b5b
commit 5b294c704e
4 changed files with 11 additions and 0 deletions

View file

@ -56,6 +56,7 @@ def completion(
if task == "conversational":
inference_params = copy.deepcopy(optional_params)
inference_params.pop("details")
inference_params.pop("return_full_text")
past_user_inputs = []
generated_responses = []
text = ""