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 . import head_detector 

18from . import loading_from_pretrained as loading 

19from . import patching 

20from . import train 

21 

22from .past_key_value_caching import ( 

23 HookedTransformerKeyValueCache as EasyTransformerKeyValueCache, 

24 HookedTransformerKeyValueCacheEntry as EasyTransformerKeyValueCacheEntry, 

25) 

26from .HookedTransformer import HookedTransformer as EasyTransformer 

27from .HookedTransformerConfig import HookedTransformerConfig as EasyTransformerConfig