Implemented transformer model families and presets.
Modules
| Module | Public API |
|---|---|
olm.models | GPT2, GPT2Large, GPT2Medium, GPT2Model, GPT2XL, Gemma2Model, Gemma2_27B, Gemma2_2B, +28 more |
olm.models.alibaba | Qwen2Model, Qwen2_5_0_5B, Qwen2_5_14B, Qwen2_5_1_5B, Qwen2_5_32B, Qwen2_5_3B, Qwen2_5_72B, Qwen2_5_7B |
olm.models.alibaba.qwen2 | Qwen2Block, Qwen2Model, Qwen2_5_0_5B, Qwen2_5_14B, Qwen2_5_1_5B, Qwen2_5_32B, Qwen2_5_3B, Qwen2_5_72B, +1 more |
olm.models.allenai | OLMoModel, OLMo_7B |
olm.models.allenai.olmo | OLMoBlock, OLMoModel, OLMo_7B |
olm.models.facebook | OPT125M, OPTModel |
olm.models.facebook.opt | OPT125M, OPTBlock, OPTModel |
olm.models.google | Gemma2Model, Gemma2_27B, Gemma2_2B, Gemma2_9B |
olm.models.google.gemma2 | Gemma2Block, Gemma2Embedding, Gemma2FinalLogitSoftcap, Gemma2Model, Gemma2_27B, Gemma2_2B, Gemma2_9B |
olm.models.meta | Llama2Model, Llama2_13B, Llama2_70B, Llama2_7B, Llama3Model, Llama3_1_405B, Llama3_1_70B, Llama3_1_8B, +2 more |
olm.models.meta.llama2 | Llama2Block, Llama2Model, Llama2_13B, Llama2_70B, Llama2_7B |
olm.models.meta.llama3 | Llama3Block, Llama3Model, Llama3_1_405B, Llama3_1_70B, Llama3_1_8B, Llama3_2_1B, Llama3_2_3B |
olm.models.microsoft | Phi3Model, Phi3_5_Mini, Phi3_Small, Phi4Model, Phi4_14B |
olm.models.microsoft.phi3 | Phi3Block, Phi3Model, Phi3_5_Mini, Phi3_Small |
olm.models.microsoft.phi4 | Phi4Block, Phi4Model, Phi4_14B |
olm.models.openai | GPT2, GPT2Large, GPT2Medium, GPT2Model, GPT2XL |
olm.models.openai.gpt2 | GPT2, GPT2Block, GPT2Large, GPT2Medium, GPT2Model, GPT2XL |