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

@@ -1,11 +1,7 @@
# coding=utf-8
# Adapted from
# https://github.com/THUDM/ChatGLM2-6B
"""Inference-only ChatGLM model compatible with THUDM 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 ChatGLM model compatible with THUDM weights."""
from typing import List, Optional, Tuple
import torch