pyreft icon indicating copy to clipboard operation
pyreft copied to clipboard

[P1] Intervention Locations more than Prefix and Suffix

Open comeandcode opened this issue 7 months ago • 5 comments

Does the backbone pyvene support more than two intervention blocks on one layer? I met anaconda3/envs/reft_train/lib/python3.10/site-packages/pyvene/models/intervenable_base.py", line 1092, in _wait_for_forward_with_parallel_intervention unit_locations_base[ IndexError: list index out of range when I tried to add the third intervention block into one layer.

comeandcode avatar Jul 24 '24 22:07 comeandcode