diff --git a/docs/source/models/supported_models.rst b/docs/source/models/supported_models.rst index 10d22ff461fb5..2663cf2366e64 100644 --- a/docs/source/models/supported_models.rst +++ b/docs/source/models/supported_models.rst @@ -70,7 +70,7 @@ Alongside each architecture, we include some popular models that use it. - :code:`Qwen/Qwen-7B`, :code:`Qwen/Qwen-7B-Chat`, etc. * - :code:`Qwen2ForCausalLM` - Qwen2 - - :code:`Qwen/Qwen2-7B-beta`, :code:`Qwen/Qwen-7B-Chat-beta`, etc. + - :code:`Qwen/Qwen2-beta-7B`, :code:`Qwen/Qwen2-beta-7B-Chat`, etc. * - :code:`StableLMEpochForCausalLM` - StableLM - :code:`stabilityai/stablelm-3b-4e1t/` , :code:`stabilityai/stablelm-base-alpha-7b-v2`, etc. diff --git a/vllm/model_executor/models/qwen2.py b/vllm/model_executor/models/qwen2.py index ee6c9f54a2dd5..e823e6f8c3dbe 100644 --- a/vllm/model_executor/models/qwen2.py +++ b/vllm/model_executor/models/qwen2.py @@ -1,7 +1,8 @@ # coding=utf-8 # Adapted from # https://github.com/huggingface/transformers/blob/v4.28.0/src/transformers/models/qwen2/modeling_qwen2.py -# Copyright 2024 The Qwen team and the HuggingFace Inc. team. +# Copyright 2024 The Qwen team. +# Copyright 2023 The vLLM team. # Copyright 2022 EleutherAI and the HuggingFace Inc. team. All rights reserved. # # This code is based on EleutherAI's GPT-NeoX library and the GPT-NeoX @@ -20,7 +21,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Inference-only Mistral model compatible with HuggingFace weights.""" +"""Inference-only Qwen2 model compatible with HuggingFace weights.""" from typing import List, Optional, Tuple import torch