Coverage for transformer_lens/model_bridge/supported_architectures/olmo3.py: 100%

1"""OLMo 3 architecture adapter.""" 

2 

3from transformer_lens.model_bridge.supported_architectures.olmo2 import ( 

4 Olmo2ArchitectureAdapter, 

5) 

6 

7 

8class Olmo3ArchitectureAdapter(Olmo2ArchitectureAdapter): 

9 """Architecture adapter for OLMo 3 / OLMo 3.1 models. 

10 

11 OLMo 3 is architecturally identical to OLMo 2 at the weight and component level. 

12 The only difference is sliding window attention on some layers (configurable via 

13 layer_types), which is handled by the HF model's forward pass (mask creation) 

14 and does not affect weight structure or component mapping. 

15 """ 

16 

17 pass