bigcode
/

santacoder

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

lvwerra HF staff commited on Dec 2, 2022

Commit

55a1ff8

•

1 Parent(s): 5fb20b0

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 tools.hf_transformers.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