Skip to content

GitLab

Commit d5723fbb authored by zhuwenwen's avatar zhuwenwen
Browse files

update qwen2 supported_lora_modules

parent 41b09879
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/decoder_only/vision_language/test_internvl.py
View file @ d5723fbb
...@@ -366,6 +366,7 @@ def test_different_num_patches(hf_runner, vllm_runner, image_assets, model, ...@@ -366,6 +366,7 @@ def test_different_num_patches(hf_runner, vllm_runner, image_assets, model,
) )
@pytest.mark.skipif(is_hip(), reason="InternVL2-AWQ is not supported on ROCm.")
@pytest.mark.parametrize( @pytest.mark.parametrize(
"models", [(os.path.join(models_path_prefix, "OpenGVLab/InternVL2-2B"), os.path.join(models_path_prefix, "OpenGVLab/InternVL2-2B-AWQ") if not is_hip() else [])]) "models", [(os.path.join(models_path_prefix, "OpenGVLab/InternVL2-2B"), os.path.join(models_path_prefix, "OpenGVLab/InternVL2-2B-AWQ") if not is_hip() else [])])
@pytest.mark.parametrize( @pytest.mark.parametrize(
......
vllm/model_executor/models/qwen2.py
View file @ d5723fbb
...@@ -329,7 +329,6 @@ class Qwen2ForCausalLM(nn.Module, SupportsLoRA): ...@@ -329,7 +329,6 @@ class Qwen2ForCausalLM(nn.Module, SupportsLoRA):
"o_proj", "o_proj",
"gate_up_proj", "gate_up_proj",
"down_proj", "down_proj",
"lm_head"
] ]
embedding_modules = {} embedding_modules = {}
embedding_padding_modules = [] embedding_padding_modules = []
......
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