bigcode
/

santacoder

Text Generation

text-generation-inference

Model card Files Files and versions

lvwerra HF Staff commited on Dec 2, 2022

Commit

094f6ad

·

1 Parent(s): 55a1ff8

Update modeling_gpt2_mq.py

Files changed (1) hide show

modeling_gpt2_mq.py +1 -1

modeling_gpt2_mq.py CHANGED Viewed

@@ -32,7 +32,7 @@ from transformers.utils import (
 )
 from transformers.utils.model_parallel_utils import assert_device_map, get_device_map
 from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
-from configuration_gpt2_mq import GPT2CustomConfig, MULTI_QUERY, MULTI_HEAD

 )
 from transformers.utils.model_parallel_utils import assert_device_map, get_device_map
 from transformers.models.gpt2.modeling_gpt2 import GPT2Model, GPT2Block, GPT2PreTrainedModel, GPT2LMHeadModel
+from .configuration_gpt2_mq import GPT2CustomConfig, MULTI_QUERY, MULTI_HEAD