fix hf conversational task bug

This commit is contained in:
Krrish Dholakia 2023-09-18 11:46:14 -07:00
parent 3920db080f
commit 78d579b04b
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 = ""