Coverage for transformer_lens/pretrained/weight_conversions/__init__.py: 100%

20 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2024-10-04 23:19 +0000

1from .neo import convert_neo_weights 

2from .gpt2 import convert_gpt2_weights 

3from .opt import convert_opt_weights 

4from .gptj import convert_gptj_weights 

5from .neox import convert_neox_weights 

6from .llama import convert_llama_weights 

7from .bert import convert_bert_weights 

8from .mistral import convert_mistral_weights 

9from .mixtral import convert_mixtral_weights 

10from .bloom import convert_bloom_weights 

11from .coder import convert_coder_weights 

12from .qwen import convert_qwen_weights 

13from .qwen2 import convert_qwen2_weights 

14from .phi import convert_phi_weights 

15from .phi3 import convert_phi3_weights 

16from .gemma import convert_gemma_weights 

17from .mingpt import convert_mingpt_weights 

18from .nanogpt import convert_nanogpt_weights 

19from .t5 import convert_t5_weights 

20from .neel_solu_old import convert_neel_solu_old_weights