Fix model docstrings (#1764)

This commit is contained in:
Woosuk Kwon
2023-11-23 23:04:44 -08:00
committed by GitHub
parent e0c6f556e8
commit 7c600440f7
14 changed files with 14 additions and 70 deletions

View File

@@ -3,11 +3,7 @@
# https://huggingface.co/Qwen/Qwen-7B/blob/main/modeling_qwen.py
# Copyright (c) Alibaba Cloud.
# LICENSE: https://huggingface.co/Qwen/Qwen-7B/blob/main/LICENSE
"""Inference-only QWen model compatible with HuggingFace weights.
The input of the model is flattened to a 1D tensor of tokens. The model uses
InputMetadata to extract the original 2D shape of the input.
"""
"""Inference-only QWen model compatible with HuggingFace weights."""
from typing import Any, Dict, List, Optional, Tuple
import torch