Skip to content

Commit 2832e7b

Browse files
authored
fix names and license for Qwen2 (#2589)
1 parent 3a7dd7e commit 2832e7b

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

docs/source/models/supported_models.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ Alongside each architecture, we include some popular models that use it.
7070
- :code:`Qwen/Qwen-7B`, :code:`Qwen/Qwen-7B-Chat`, etc.
7171
* - :code:`Qwen2ForCausalLM`
7272
- Qwen2
73-
- :code:`Qwen/Qwen2-7B-beta`, :code:`Qwen/Qwen-7B-Chat-beta`, etc.
73+
- :code:`Qwen/Qwen2-beta-7B`, :code:`Qwen/Qwen2-beta-7B-Chat`, etc.
7474
* - :code:`StableLMEpochForCausalLM`
7575
- StableLM
7676
- :code:`stabilityai/stablelm-3b-4e1t/` , :code:`stabilityai/stablelm-base-alpha-7b-v2`, etc.

vllm/model_executor/models/qwen2.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
# coding=utf-8
22
# Adapted from
33
# https://github.com/huggingface/transformers/blob/v4.28.0/src/transformers/models/qwen2/modeling_qwen2.py
4-
# Copyright 2024 The Qwen team and the HuggingFace Inc. team.
4+
# Copyright 2024 The Qwen team.
5+
# Copyright 2023 The vLLM team.
56
# Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved.
67
#
78
# This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX
@@ -20,7 +21,7 @@
2021
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
2122
# See the License for the specific language governing permissions and
2223
# limitations under the License.
23-
"""Inference-only Mistral model compatible with HuggingFace weights."""
24+
"""Inference-only Qwen2 model compatible with HuggingFace weights."""
2425
from typing import List, Optional, Tuple
2526

2627
import torch

0 commit comments

Comments
 (0)