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

1"""Sources module. 

2 

3This module provides functionality to load and convert models from HuggingFace to TransformerLens format. 

4""" 

5 

6from transformer_lens.model_bridge.sources._bridge_builder import ( 

7 build_bridge_config_from_hf, 

8 build_bridge_from_module, 

9 detect_tokenizer_bos_eos, 

10) 

11from transformer_lens.model_bridge.sources.transformers import ( 

12 boot, 

13 check_model_support, 

14 list_supported_models, 

15) 

16 

17__all__ = [ 

18 "boot", 

19 "build_bridge_config_from_hf", 

20 "build_bridge_from_module", 

21 "check_model_support", 

22 "detect_tokenizer_bos_eos", 

23 "list_supported_models", 

24]