[Bug] Visualization code broken

dtch1997 / sae-eap

Edge attribution patching with SAEs

1 stars 0 forks source link

[Bug] Visualization code broken #6

Open dtch1997 opened 3 months ago

dtch1997 commented 3 months ago

The visualization code used to depend on Edge.type but that field has been removed. We need a new way to determine the type of edge (q, k, v, or other).

This could be done via a function: get_edge_type(edge: Edge).