olivierdehaene
/

optimized-santacoder

Text Generation

text-generation-inference

Model card Files Files and versions Community

olivierdehaene HF staff commited on Jan 19, 2023

Commit

02857b0

·

1 Parent(s): 9b6553f

Update modeling_gpt2_mq.py

Files changed (1) hide show

modeling_gpt2_mq.py +2 -1

modeling_gpt2_mq.py CHANGED Viewed

@@ -13,7 +13,8 @@ from transformers.modeling_outputs import (
 )
 from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
 from transformers.utils import logging
-from configuration_gpt2_mq import GPT2CustomConfig, MULTI_QUERY
 logger = logging.get_logger(__name__)

 )
 from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
 from transformers.utils import logging
+from .configuration_gpt2_mq import GPT2CustomConfig, MULTI_QUERY
 logger = logging.get_logger(__name__)