Small bug fix

lukaemon · Sep 27, 2023 · 8544819 · 8544819
1 parent c18d5b9
commit 8544819
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -107,7 +107,7 @@ def interactive(model_path: str, max_tokens: int = 35, temperature: float = 0.7)
 
     while True:
         prompt = input("Prompt: ")
-        res, _logprobs = generate([prompt], transformer, tokenizer, max_tokens)
+        res, _logprobs = generate([prompt], transformer, tokenizer, max_tokens=max_tokens)
         print(res[0])
         print("=====================")