File tree Expand file tree Collapse file tree 3 files changed +4
-4
lines changed
tests/models/language/pooling
vllm/model_executor/models Expand file tree Collapse file tree 3 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 33
33
enable_test = True ),
34
34
EmbedModelInfo ("Alibaba-NLP/gte-base-en-v1.5" ,
35
35
architecture = "GteNewModel" ,
36
- enable_test = True ),
36
+ enable_test = False ),
37
37
EmbedModelInfo ("Alibaba-NLP/gte-large-en-v1.5" ,
38
38
architecture = "GteNewModel" ,
39
- enable_test = True ),
39
+ enable_test = False ),
40
40
########### Qwen2ForCausalLM
41
41
EmbedModelInfo ("Alibaba-NLP/gte-Qwen2-1.5B-instruct" ,
42
42
architecture = "Qwen2ForCausalLM" ,
Original file line number Diff line number Diff line change @@ -209,7 +209,7 @@ def __init__(
209
209
prefix = maybe_prefix (
210
210
prefix , "score" ))
211
211
212
- def config_verify (self , vllm_config ):
212
+ def verify_and_update_config (self , vllm_config ):
213
213
# Leave an interface for validating and modifying model_config
214
214
# for slightly different models
215
215
pass
Original file line number Diff line number Diff line change @@ -331,7 +331,7 @@ def __init__(
331
331
) -> None :
332
332
super ().__init__ (vllm_config = vllm_config , prefix = prefix )
333
333
334
- def config_verify (self , vllm_config : "VllmConfig" ):
334
+ def verify_and_update_config (self , vllm_config : "VllmConfig" ):
335
335
config = vllm_config .model_config .hf_config
336
336
337
337
is_original_qwen3_reranker = getattr (config ,
You can’t perform that action at this time.
0 commit comments