fix(activation): preserve tokens type during dtype conversion

Hzfinfdu · Hzfinfdu · commit 6b84c492505e · 2025-02-15T00:16:10.000+08:00
diff --git a/src/lm_saes/activation/processors/cached_activation.py b/src/lm_saes/activation/processors/cached_activation.py
@@ -261,4 +261,4 @@ def __getitem__(self, chunk_idx):
         return self.activation_loader.load_chunk_for_hooks(
             chunk_idx,
             self.hook_chunks,
-        )
+        )
