Coverage for transformer_lens/conversion_utils/conversion_steps/__init__.py: 100%

1"""Architecture adapter conversion steps. 

2 

3This module contains the conversion steps for converting between different model architectures. 

4""" 

5 

6from transformer_lens.conversion_utils.conversion_steps.arithmetic_tensor_conversion import ( 

7 ArithmeticTensorConversion, 

8) 

9from transformer_lens.conversion_utils.conversion_steps.attention_auto_conversion import ( 

10 AttentionAutoConversion, 

11) 

12from transformer_lens.conversion_utils.conversion_steps.base_tensor_conversion import ( 

13 BaseTensorConversion, 

14) 

15from transformer_lens.conversion_utils.conversion_steps.callable_tensor_conversion import ( 

16 CallableTensorConversion, 

17) 

18from transformer_lens.conversion_utils.conversion_steps.chain_tensor_conversion import ( 

19 ChainTensorConversion, 

20) 

21from transformer_lens.conversion_utils.conversion_steps.tensor_conversion_set import ( 

22 TensorConversionSet, 

23) 

24from transformer_lens.conversion_utils.conversion_steps.rearrange_tensor_conversion import ( 

25 RearrangeTensorConversion, 

26) 

27from transformer_lens.conversion_utils.conversion_steps.repeat_tensor_conversion import ( 

28 RepeatTensorConversion, 

29) 

30from transformer_lens.conversion_utils.conversion_steps.split_tensor_conversion import ( 

31 SplitTensorConversion, 

32) 

33from transformer_lens.conversion_utils.conversion_steps.ternary_tensor_conversion import ( 

34 TernaryTensorConversion, 

35) 

36from transformer_lens.conversion_utils.conversion_steps.transpose_tensor_conversion import ( 

37 TransposeTensorConversion, 

38) 

39from transformer_lens.conversion_utils.conversion_steps.zeros_like_conversion import ( 

40 ZerosLikeConversion, 

41) 

42 

43__all__ = [ 

44 "ArithmeticTensorConversion", 

45 "AttentionAutoConversion", 

46 "BaseTensorConversion", 

47 "CallableTensorConversion", 

48 "ChainTensorConversion", 

49 "RearrangeTensorConversion", 

50 "RepeatTensorConversion", 

51 "SplitTensorConversion", 

52 "TernaryTensorConversion", 

53 "TensorConversionSet", 

54 "TransposeTensorConversion", 

55 "ZerosLikeConversion", 

56]