Coverage for transformer_lens/__init__.py: 89%

1from . import ( 

2 components, 

3 conversion_utils, 

4 evals, 

5 factories, 

6 head_detector, 

7 hook_points, 

8 patching, 

9 train, 

10 utilities, 

11) 

12from . import loading_from_pretrained as loading 

13from . import supported_models 

14from .ActivationCache import ActivationCache 

15from .BertNextSentencePrediction import BertNextSentencePrediction 

16from .cache.key_value_cache import TransformerLensKeyValueCache 

17from .cache.key_value_cache_entry import TransformerLensKeyValueCacheEntry 

18from .config import HookedTransformerConfig 

19from .FactoredMatrix import FactoredMatrix 

20from .HookedEncoder import HookedEncoder 

21from .HookedAudioEncoder import HookedAudioEncoder 

22from .HookedEncoderDecoder import HookedEncoderDecoder 

23from .HookedTransformer import HookedTransformer 

24 

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

26try: 

27 from . import lit 

28except ImportError: 

29 # LIT is an optional dependency 

30 lit = None # type: ignore 

31 

32from .SVDInterpreter import SVDInterpreter 

33 

34__all__ = [ 

35 "HookedTransformerConfig", 

36 "FactoredMatrix", 

37 "ActivationCache", 

38 "HookedTransformer", 

39 "SVDInterpreter", 

40 "HookedEncoder", 

41 "HookedEncoderDecoder", 

42 "TransformerLensKeyValueCache", 

43 "TransformerLensKeyValueCacheEntry", 

44 "components", 

45 "conversion_utils", 

46 "factories", 

47 "utilities", 

48]