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

@@ -17,11 +17,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 GPTBigCode 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 GPTBigCode model compatible with HuggingFace weights."""
from typing import List, Optional, Tuple
import torch