Commit
·
02857b0
1
Parent(s):
9b6553f
Update modeling_gpt2_mq.py
Browse files- modeling_gpt2_mq.py +2 -1
modeling_gpt2_mq.py
CHANGED
|
@@ -13,7 +13,8 @@ from transformers.modeling_outputs import (
|
|
| 13 |
)
|
| 14 |
from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
|
| 15 |
from transformers.utils import logging
|
| 16 |
-
|
|
|
|
| 17 |
|
| 18 |
logger = logging.get_logger(__name__)
|
| 19 |
|
|
|
|
| 13 |
)
|
| 14 |
from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
|
| 15 |
from transformers.utils import logging
|
| 16 |
+
|
| 17 |
+
from .configuration_gpt2_mq import GPT2CustomConfig, MULTI_QUERY
|
| 18 |
|
| 19 |
logger = logging.get_logger(__name__)
|
| 20 |
|