Skip to content

Commit e8f369a

Browse files
Isotr0pyMu Huai
authored and
Mu Huai
committed
[Misc] Remove OLMo2 config copy (vllm-project#17066)
Signed-off-by: Isotr0py <2037008807@qq.com> Signed-off-by: Mu Huai <tianbowen.tbw@antgroup.com>
1 parent 971cf4c commit e8f369a

File tree

4 files changed

+4
-176
lines changed

4 files changed

+4
-176
lines changed

vllm/model_executor/models/olmo2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828

2929
import torch
3030
from torch import nn
31+
from transformers import Olmo2Config
3132

3233
from vllm.attention import Attention
3334
from vllm.config import VllmConfig
@@ -51,7 +52,6 @@
5152
make_layers, maybe_prefix)
5253
from vllm.model_executor.sampling_metadata import SamplingMetadata
5354
from vllm.sequence import IntermediateTensors
54-
from vllm.transformers_utils.configs.olmo2 import Olmo2Config
5555

5656

5757
class Olmo2Attention(nn.Module):

vllm/transformers_utils/config.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,9 @@
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
4443
from vllm.transformers_utils.utils import check_gguf_file
4544
from vllm.utils import resolve_obj_by_qualname
@@ -76,7 +75,6 @@
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,

vllm/transformers_utils/configs/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from vllm.transformers_utils.configs.mpt import MPTConfig
2222
from vllm.transformers_utils.configs.nemotron import NemotronConfig
2323
from vllm.transformers_utils.configs.nvlm_d import NVLM_D_Config
24-
from vllm.transformers_utils.configs.olmo2 import Olmo2Config
2524
from vllm.transformers_utils.configs.skyworkr1v import SkyworkR1VChatConfig
2625
from vllm.transformers_utils.configs.solar import SolarConfig
2726
from vllm.transformers_utils.configs.telechat2 import Telechat2Config
@@ -46,7 +45,6 @@
4645
"KimiVLConfig",
4746
"NemotronConfig",
4847
"NVLM_D_Config",
49-
"Olmo2Config",
5048
"SkyworkR1VChatConfig",
5149
"SolarConfig",
5250
"Telechat2Config",

vllm/transformers_utils/configs/olmo2.py

Lines changed: 0 additions & 168 deletions
This file was deleted.

0 commit comments

Comments
 (0)