Open ocramz opened 5 months ago
The first example in the README could use some introduction to make sense of the interface, e.g :
b
s
zeroout_intervention_fn
intervention
def zeroout_intervention_fn(b, s): b[:,3] = 0. # 3rd position return b pv_model = pv.IntervenableModel({ "component": "model.layers[15].mlp.output", # string access "intervention": zeroout_intervention_fn}, model=model)
Suggestion / Feature Request
The first example in the README could use some introduction to make sense of the interface, e.g :
b
ands
ofzeroout_intervention_fn
?intervention
field of the configuration dict always a binary function?