Coverage for transformer_lens/model_bridge/sources/native/__init__.py: 100%

1"""TL-native model source for TransformerBridge.""" 

2from transformer_lens.model_bridge.sources.native.init import initialize_native_model 

3from transformer_lens.model_bridge.sources.native.model import ( 

4 NativeAttention, 

5 NativeBlock, 

6 NativeMLP, 

7 NativeModel, 

8) 

9 

10__all__ = [ 

11 "NativeAttention", 

12 "NativeBlock", 

13 "NativeMLP", 

14 "NativeModel", 

15 "initialize_native_model", 

16]