Coverage for transformer_lens/__init__.py: 92%

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

17from .HookedEncoderDecoder import HookedEncoderDecoder 

18from .BertNextSentencePrediction import BertNextSentencePrediction 

19from . import head_detector 

20from . import loading_from_pretrained as loading 

21from . import patching 

22from . import train 

23 

24# LIT integration (optional, requires lit-nlp package) 

25try: 

26 from . import lit 

27except ImportError: 

28 # LIT is an optional dependency 

29 lit = None # type: ignore 

30 

31from .past_key_value_caching import ( 

32 HookedTransformerKeyValueCache as EasyTransformerKeyValueCache, 

33 HookedTransformerKeyValueCacheEntry as EasyTransformerKeyValueCacheEntry, 

34) 

35from .HookedTransformer import HookedTransformer as EasyTransformer 

36from .HookedTransformerConfig import HookedTransformerConfig as EasyTransformerConfig