Spaces:

Th3BossC
/

ChitChat

Runtime error

Th3BossC commited on Jun 19, 2023

Commit

e8e4569

1 Parent(s): 79b7e57

fixed minor bugs

Files changed (2) hide show

ChitChat/common/utils.py CHANGED Viewed

@@ -45,7 +45,7 @@ def conversation(user, userInput):
     chat_history_ids = small_model.generate(
         bot_input_ids,
         max_length = 500,
-        pad_token_id = tokenizer.eos_token_id,
         no_repeat_ngram_size = 3,
         do_sample = True,
         top_k = 100,

     chat_history_ids = small_model.generate(
         bot_input_ids,
         max_length = 500,
+        pad_token_id = small_tokenizer.eos_token_id,
         no_repeat_ngram_size = 3,
         do_sample = True,
         top_k = 100,

instance/site.db CHANGED Viewed

Binary files a/instance/site.db and b/instance/site.db differ