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.native import boot as boot_native 

12from transformer_lens.model_bridge.sources.tl_legacy import boot as boot_tl_legacy 

13from transformer_lens.model_bridge.sources.transformers import ( 

14 boot, 

15 check_model_support, 

16 list_supported_models, 

17) 

18from transformer_lens.model_bridge.sources.vllm import boot_vllm 

19 

20__all__ = [ 

21 "boot", 

22 "boot_native", 

23 "boot_tl_legacy", 

24 "boot_vllm", 

25 "build_bridge_config_from_hf", 

26 "build_bridge_from_module", 

27 "check_model_support", 

28 "detect_tokenizer_bos_eos", 

29 "list_supported_models", 

30]