refactor: fix previous refactors

2023-05-28 22:25:42 +09:00
parent 8bd7a49cd7
commit 56f9ca5709
3 changed files with 4 additions and 4 deletions
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -83,7 +83,7 @@ def do_inference(cfg, model, tokenizer, prompter="AlpacaPrompter"):
                temperature=0.9,
                top_p=0.95,
                top_k=40,
-                return_DictDefault_in_generate=True,
+                return_dict_in_generate=True,
                output_attentions=False,
                output_hidden_states=False,
                output_scores=False,