ignore type eror

jbloomAus · Mar 28, 2024 · e87198b · e87198b
1 parent 67dfb46
commit e87198b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sae_training/activations_store.py b/sae_training/activations_store.py
@@ -346,7 +346,7 @@ def _get_next_dataset_tokens(self) -> torch.Tensor:
             )
             if (
                 not self.cfg.prepend_bos
-                and tokens[0] == self.model.tokenizer.bos_token_id
+                and tokens[0] == self.model.tokenizer.bos_token_id  # type: ignore
             ):
                 tokens = tokens[1:]
         return tokens