Types & classes — github.com/boyiwei/alignment-attribution-code

CodeHub MCPcopy Create free account

hub / github.com/boyiwei/alignment-attribution-code / types & classes

Types & classes14 in github.com/boyiwei/alignment-attribution-code

⨍ Functions 99 ◇ Types & classes 14

↓ 5 callersClassWrappedGPT

This class wraps a GPT layer for specific operations.

lib/layerwrapper.py:6

↓ 3 callersClassCatcher

lib/prune.py:110

↓ 3 callersClassno_act_recording

lib/model_wrapper.py:60

↓ 2 callersClassset_mask

main_low_rank_diff.py:61

↓ 1 callersClassAblateGPT

lib/ablate.py:12

↓ 1 callersClassActLinear

drop in replacement of nn.Linear

main_low_rank_diff.py:18

↓ 1 callersClassActLinear

drop in replacement of nn.Linear

lib/model_wrapper.py:16

↓ 1 callersClassActLinear

drop in replacement of nn.Linear

lib/model_wrapper_low.py:17

↓ 1 callersClassSparseGPT

lib/sparsegpt.py:13

↓ 1 callersClassno_act_recording

main_low_rank_diff.py:46

↓ 1 callersClassno_act_recording

lib/model_wrapper_low.py:45

↓ 1 callersClassset_mask

lib/model_wrapper.py:75

↓ 1 callersClassset_mask

lib/model_wrapper_low.py:60

ClassTokenizerWrapper