[Model] Add support for GPT-J (#226)

Co-authored-by: woWoosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Andre Slavescu
2023-07-08 20:55:16 -04:00
committed by GitHub
parent 75beba29b5
commit c894836108
10 changed files with 269 additions and 7 deletions

View File

@@ -1,3 +1,4 @@
# coding=utf-8
# Adapted from https://huggingface.co/mosaicml/mpt-7b/tree/main
import math
from typing import Dict, List, Optional, Tuple