transformer_lensΒΆ
- transformer_lens package
- Subpackages
- transformer_lens.benchmarks package
- transformer_lens.cache package
- transformer_lens.config package
- transformer_lens.conversion_utils package
- transformer_lens.factories package
- transformer_lens.lit package
- transformer_lens.model_bridge package
- transformer_lens.tools package
- transformer_lens.utilities package
- Submodules
- transformer_lens.HookedRootModule module
- transformer_lens.evals module
- transformer_lens.head_detector module
- transformer_lens.hook_points module
- transformer_lens.model_protocol module
- transformer_lens.patching module
PatchingResultgeneric_activation_patch()get_act_patch_attn_head_all_pos_every()get_act_patch_attn_head_by_pos_every()get_act_patch_attn_head_k_all_pos()get_act_patch_attn_head_k_by_pos()get_act_patch_attn_head_out_all_pos()get_act_patch_attn_head_out_by_pos()get_act_patch_attn_head_pattern_all_pos()get_act_patch_attn_head_pattern_by_pos()get_act_patch_attn_head_pattern_dest_src_pos()get_act_patch_attn_head_q_all_pos()get_act_patch_attn_head_q_by_pos()get_act_patch_attn_head_v_all_pos()get_act_patch_attn_head_v_by_pos()get_act_patch_attn_out()get_act_patch_block_every()get_act_patch_mlp_out()get_act_patch_resid_mid()get_act_patch_resid_pre()layer_head_dest_src_pos_pattern_patch_setter()layer_head_pattern_patch_setter()layer_head_pos_pattern_patch_setter()layer_head_vector_patch_setter()layer_pos_head_vector_patch_setter()layer_pos_patch_setter()
- transformer_lens.supported_models module
- transformer_lens.weight_processing module
- Module contents
HookedRootModuleHookedRootModule.add_caching_hooks()HookedRootModule.add_hook()HookedRootModule.add_perma_hook()HookedRootModule.cache_all()HookedRootModule.cache_some()HookedRootModule.check_and_add_hook()HookedRootModule.check_hooks_to_add()HookedRootModule.clear_contexts()HookedRootModule.get_caching_hooks()HookedRootModule.hook_dictHookedRootModule.hook_points()HookedRootModule.hooks()HookedRootModule.mod_dictHookedRootModule.nameHookedRootModule.remove_all_hook_fns()HookedRootModule.reset_hooks()HookedRootModule.run_with_cache()HookedRootModule.run_with_hooks()HookedRootModule.setup()
TransformerBridgeTransformerBridge.OVTransformerBridge.OV_for_attn_layers()TransformerBridge.QKTransformerBridge.QK_for_attn_layers()TransformerBridge.W_ETransformerBridge.W_E_posTransformerBridge.W_KTransformerBridge.W_OTransformerBridge.W_QTransformerBridge.W_UTransformerBridge.W_VTransformerBridge.W_gateTransformerBridge.W_inTransformerBridge.W_outTransformerBridge.W_posTransformerBridge.__init__()TransformerBridge.accumulated_bias()TransformerBridge.all_composition_scores()TransformerBridge.all_head_labelsTransformerBridge.attn_head_labelsTransformerBridge.b_KTransformerBridge.b_OTransformerBridge.b_QTransformerBridge.b_UTransformerBridge.b_VTransformerBridge.b_inTransformerBridge.b_outTransformerBridge.block_hooks()TransformerBridge.block_submodules()TransformerBridge.blocks_with()TransformerBridge.boot_native()TransformerBridge.boot_tl_legacy()TransformerBridge.boot_transformers()TransformerBridge.check_model_support()TransformerBridge.composition_layer_indices()TransformerBridge.cpu()TransformerBridge.cuda()TransformerBridge.diffusion_generate()TransformerBridge.enable_compatibility_mode()TransformerBridge.encoder_output()TransformerBridge.forward()TransformerBridge.generate()TransformerBridge.generate_stream()TransformerBridge.get_params()TransformerBridge.get_token_position()TransformerBridge.hf_generate()TransformerBridge.init_weights()TransformerBridge.input_to_embed()TransformerBridge.layer_types()TransformerBridge.list_supported_models()TransformerBridge.load_state_dict()TransformerBridge.mps()TransformerBridge.n_params_totalTransformerBridge.named_parameters()TransformerBridge.original_modelTransformerBridge.parameters()TransformerBridge.predict_next_sentence()TransformerBridge.prepare_multimodal_inputs()TransformerBridge.process_weights()TransformerBridge.set_use_attn_in()TransformerBridge.set_use_attn_result()TransformerBridge.set_use_hook_mlp_in()TransformerBridge.set_use_split_qkv_input()TransformerBridge.stack_params_for()TransformerBridge.state_dict()TransformerBridge.tl_named_parameters()TransformerBridge.tl_parameters()TransformerBridge.to()TransformerBridge.to_sentence_pair_tokens()TransformerBridge.to_single_str_token()TransformerBridge.to_single_token()TransformerBridge.to_str_tokens()TransformerBridge.to_string()TransformerBridge.to_tokens()TransformerBridge.tokens_to_residual_directions()TransformerBridge.train()
TransformerBridgeConfigTransformerLensKeyValueCacheTransformerLensKeyValueCache.__getitem__()TransformerLensKeyValueCache.append_attention_mask()TransformerLensKeyValueCache.entriesTransformerLensKeyValueCache.freeze()TransformerLensKeyValueCache.frozenTransformerLensKeyValueCache.init_cache()TransformerLensKeyValueCache.previous_attention_maskTransformerLensKeyValueCache.unfreeze()
TransformerLensKeyValueCacheEntry
- Subpackages