Index

A

activation_extractor.model_functions.default_hooked_layers module activation_extractor.model_functions.embedding_to_numpy module activation_extractor.model_functions.inference_funs module

activation_extractor.model_functions.load_models module activation_extractor.model_functions.model_types module activation_extractor.model_functions.tokenize_funs module

C

clear_all_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)

create_hook() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)

D

define_inference_function() (in module activation_extractor.model_functions.inference_funs)

define_tokenize_function() (in module activation_extractor.model_functions.tokenize_funs) detach_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)

E

emb_reformatting() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)

embedding_to_numpy() (in module activation_extractor.model_functions.embedding_to_numpy)

G

get_layers_to_hook() (in module activation_extractor.model_functions.default_hooked_layers)

get_outputs() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method) gpu_to_cpu() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)

I

IntermediateExtractor (class in activation_extractor.extractors.intermediateExtractor)

IntermediateExtractorBase (class in activation_extractor.extractors.intermediateExtractorBase)

L

load_model() (in module activation_extractor.model_functions.load_models)

load_tokenizer() (in module activation_extractor.model_functions.load_models)

M

module activation_extractor.model_functions.default_hooked_layers activation_extractor.model_functions.embedding_to_numpy activation_extractor.model_functions.inference_funs activation_extractor.model_functions.load_models activation_extractor.model_functions.model_types activation_extractor.model_functions.tokenize_funs

R

register_hooks() (activation_extractor.extractors.intermediateExtractorBase.IntermediateExtractorBase method)

S

save_outputs() (activation_extractor.extractors.intermediateExtractor.IntermediateExtractor method)