Commit 469a3a88 authored by zhuwenwen's avatar zhuwenwen
Browse files

update version

parent f51086de
...@@ -488,11 +488,11 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -488,11 +488,11 @@ def get_version_add(sha: Optional[str] = None) -> str:
if sha is None: if sha is None:
sha = get_sha(vllm_root) sha = get_sha(vllm_root)
if (major, minor) == ('2', '4'): if (major, minor) == ('2', '4'):
version = 'das.opt1.beta.' + sha[:7] version = 'das.opt2.alpha.' + sha[:7]
# version = 'das.opt1.' + sha[:7] # version = 'das.opt1.' + sha[:7]
else: else:
if (major, minor) == ('2', '4'): if (major, minor) == ('2', '4'):
version = 'das.opt1.beta' version = 'das.opt2.alpha'
# version = 'das.opt1' # version = 'das.opt1'
......
...@@ -79,10 +79,9 @@ def get_model_architecture( ...@@ -79,10 +79,9 @@ def get_model_architecture(
model_config: ModelConfig) -> Tuple[Type[nn.Module], str]: model_config: ModelConfig) -> Tuple[Type[nn.Module], str]:
architectures = getattr(model_config.hf_config, "architectures", []) architectures = getattr(model_config.hf_config, "architectures", [])
visions = getattr(model_config.hf_config, "visual", []) or getattr(model_config.hf_config, "vision_config", []) visions = getattr(model_config.hf_config, "visual", []) or getattr(model_config.hf_config, "vision_config", [])
# TODO: support deepseek distillation series models ( 'LlamaForCausalLM', 'Qwen2ForCausalLM' ) # 'Qwen2_5_VLForConditionalGeneration'
# 'Qwen2VLForConditionalGeneration', 'Qwen2_5_VLForConditionalGeneration' support_nn_architectures = ['LlamaForCausalLM', 'QWenLMHeadModel', 'Qwen2ForCausalLM', 'Qwen2MoeForCausalLM',
support_nn_architectures = ['QWenLMHeadModel', 'Qwen2VLForConditionalGeneration','ChatGLMModel', 'ChatGLMForConditionalGeneration',
'Qwen2MoeForCausalLM', 'ChatGLMModel', 'ChatGLMForConditionalGeneration',
'BaichuanForCausalLM', 'BloomForCausalLM', 'MedusaModel', 'MixtralForCausalLM', 'BaichuanForCausalLM', 'BloomForCausalLM', 'MedusaModel', 'MixtralForCausalLM',
'MLPSpeculatorPreTrainedModel', 'FalconForCausalLM', 'DeepseekV2ForCausalLM', 'MLPSpeculatorPreTrainedModel', 'FalconForCausalLM', 'DeepseekV2ForCausalLM',
'DeepseekV3ForCausalLM', 'DeepSeekMTPModel'] 'DeepseekV3ForCausalLM', 'DeepSeekMTPModel']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment