File tree Expand file tree Collapse file tree 4 files changed +4
-176
lines changed Expand file tree Collapse file tree 4 files changed +4
-176
lines changed Original file line number Diff line number Diff line change 2828
2929import torch
3030from torch import nn
31+ from transformers import Olmo2Config
3132
3233from vllm .attention import Attention
3334from vllm .config import VllmConfig
5152 make_layers , maybe_prefix )
5253from vllm .model_executor .sampling_metadata import SamplingMetadata
5354from vllm .sequence import IntermediateTensors
54- from vllm .transformers_utils .configs .olmo2 import Olmo2Config
5555
5656
5757class Olmo2Attention (nn .Module ):
Original file line number Diff line number Diff line change 3636 KimiVLConfig , MedusaConfig ,
3737 MllamaConfig , MLPSpeculatorConfig ,
3838 MPTConfig , NemotronConfig ,
39- NVLM_D_Config , Olmo2Config ,
40- RWConfig , SkyworkR1VChatConfig ,
41- SolarConfig , Telechat2Config ,
42- UltravoxConfig )
39+ NVLM_D_Config , RWConfig ,
40+ SkyworkR1VChatConfig , SolarConfig ,
41+ Telechat2Config , UltravoxConfig )
4342# yapf: enable
4443from vllm .transformers_utils .utils import check_gguf_file
4544from vllm .utils import resolve_obj_by_qualname
7675 "internvl_chat" : InternVLChatConfig ,
7776 "nemotron" : NemotronConfig ,
7877 "NVLM_D" : NVLM_D_Config ,
79- "olmo2" : Olmo2Config ,
8078 "solar" : SolarConfig ,
8179 "skywork_chat" : SkyworkR1VChatConfig ,
8280 "telechat" : Telechat2Config ,
Original file line number Diff line number Diff line change 2121from vllm .transformers_utils .configs .mpt import MPTConfig
2222from vllm .transformers_utils .configs .nemotron import NemotronConfig
2323from vllm .transformers_utils .configs .nvlm_d import NVLM_D_Config
24- from vllm .transformers_utils .configs .olmo2 import Olmo2Config
2524from vllm .transformers_utils .configs .skyworkr1v import SkyworkR1VChatConfig
2625from vllm .transformers_utils .configs .solar import SolarConfig
2726from vllm .transformers_utils .configs .telechat2 import Telechat2Config
4645 "KimiVLConfig" ,
4746 "NemotronConfig" ,
4847 "NVLM_D_Config" ,
49- "Olmo2Config" ,
5048 "SkyworkR1VChatConfig" ,
5149 "SolarConfig" ,
5250 "Telechat2Config" ,
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments