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

10from .FactoredMatrix import FactoredMatrix 

11from .ActivationCache import ActivationCache 

12from .HookedTransformer import HookedTransformer 

13from .SVDInterpreter import SVDInterpreter 

14from .HookedEncoder import HookedEncoder 

15from .HookedEncoderDecoder import HookedEncoderDecoder 

16from . import head_detector 

17from . import loading_from_pretrained as loading 

18from . import patching 

19from . import train 

20 

21from .past_key_value_caching import ( 

22 HookedTransformerKeyValueCache as EasyTransformerKeyValueCache, 

23 HookedTransformerKeyValueCacheEntry as EasyTransformerKeyValueCacheEntry, 

24) 

25from .HookedTransformer import HookedTransformer as EasyTransformer 

26from .HookedTransformerConfig import HookedTransformerConfig as EasyTransformerConfig