bug fix: setting empty token info as strings

labmlai · Oct 18, 2024 · 1a0aa78 · 1a0aa78
1 parent 98b74c8
commit 1a0aa78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/inspectus/token_viz.py b/python/inspectus/token_viz.py
@@ -12,7 +12,7 @@ def visualize_tokens(
     theme: str,
 ):
     if token_info is None:
-        token_info = [{} for _ in range(len(tokens))]
+        token_info = ["" for _ in range(len(tokens))]
 
     value_names = list(values.keys())