Fix self assignment in neuron_optimum.py (#1990)

EleutherAI · Jun 18, 2024 · bdb78d2 · bdb78d2
1 parent 72bb624
commit bdb78d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lm_eval/models/neuron_optimum.py b/lm_eval/models/neuron_optimum.py
@@ -288,7 +288,7 @@ def __init__(
 
         self.vocab_size = self.tokenizer.vocab_size
         self.tokenizer.pad_token_id = self.tokenizer.eos_token_id
-        self.add_bos_token = self.add_bos_token
+        self.add_bos_token = add_bos_token
 
         self._max_length = max_length