Spaces:

sc2582
/

decision_oaif_space2

Sleeping

sc2582 commited on Sep 7, 2024

Commit

40cd6b2

verified ·

1 Parent(s): 2cfbfa6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,6 +68,7 @@ def generate_response(messages):
         eos_token_id=[
             tokenizer.eos_token_id,
             tokenizer.convert_tokens_to_ids("<|eot_id|>"),
         ],
         pad_token_id=tokenizer.eos_token_id
     )

         eos_token_id=[
             tokenizer.eos_token_id,
             tokenizer.convert_tokens_to_ids("<|eot_id|>"),
+            tokenizer.convert_tokens_to_ids("<|assistant|>"),
         ],
         pad_token_id=tokenizer.eos_token_id
     )