machine-translation

Build error

inflaton commited on Aug 3, 2024

Commit

1a1bddc

1 Parent(s): 8434dc6

fixed bug

Files changed (1) hide show

llm_toolkit/eval.py CHANGED Viewed

@@ -101,7 +101,6 @@ def evaluate_model_with_num_shots(
             tokenizer,
             datasets["test"],
             device=device,
-            num_shots=num_shots,
             batch_size=batch_size,
             max_new_tokens=max_new_tokens,
         )

             tokenizer,
             datasets["test"],
             device=device,
             batch_size=batch_size,
             max_new_tokens=max_new_tokens,
         )