Visualization

vg_token_attention.py

@@ -355,7 +355,10 @@ def run_token_ca_visualization(
         per_term_attn_full[t] = attn_per_query
 
     if not per_term_attn_kept:
-        raise ValueError(f"None of the terms were found in the first T tokens: {terms}")
+        # raise ValueError(f"None of the terms were found in the first T tokens: {terms}")
+        print(f"[TokenCA] None of the terms were found in the first T tokens: {terms}")
+        # Return an empty dict (or whatever your function usually returns)
+        return {}
 
     # aggregate terms
     agg = None