Coverage for transformer_lens/__init__.py: 100%

1from . import hook_points 

2from . import utils 

3from . import evals 

4from .past_key_value_caching import ( 

5 HookedTransformerKeyValueCache, 

6 HookedTransformerKeyValueCacheEntry, 

7) 

8from . import components 

9from . import factories 

10from .HookedTransformerConfig import HookedTransformerConfig 

11from .FactoredMatrix import FactoredMatrix 

12from .ActivationCache import ActivationCache 

13from .HookedTransformer import HookedTransformer 

14from .SVDInterpreter import SVDInterpreter 

15from .HookedEncoder import HookedEncoder 

16from .HookedEncoderDecoder import HookedEncoderDecoder 

17from .BertNextSentencePrediction import BertNextSentencePrediction 

18from . import head_detector 

19from . import loading_from_pretrained as loading 

20from . import patching 

21from . import train 

22 

23from .past_key_value_caching import ( 

24 HookedTransformerKeyValueCache as EasyTransformerKeyValueCache, 

25 HookedTransformerKeyValueCacheEntry as EasyTransformerKeyValueCacheEntry, 

26) 

27from .HookedTransformer import HookedTransformer as EasyTransformer 

28from .HookedTransformerConfig import HookedTransformerConfig as EasyTransformerConfig