[CI/Build] Split up models tests (#10069)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2024-11-10 03:39:14 +08:00
parent b09895a618
commit 51c2e1fcef
21 changed files with 115 additions and 129 deletions
--- a/vllm/model_executor/models/internlm2_ve.py
+++ b/vllm/model_executor/models/internlm2_ve.py
@@ -164,10 +164,12 @@ class InternLM2VEForCausalLM(InternLM2ForCausalLM):
        vllm_config: VllmConfig,
        prefix: str = "",
    ) -> None:
+        super().__init__(vllm_config, prefix=prefix)
+
        config = vllm_config.model_config.hf_config
        cache_config = vllm_config.cache_config
        quant_config = vllm_config.quant_config
-        super().__init__(config, cache_config, quant_config)
+
        self.model = InternLM2VEModel(config,
                                      cache_config,
                                      quant_config,