shmelev
/

test_custom_model

Inference Endpoints

Model card Files Files and versions Community

test_custom_model / configuration_bert.py

shmelev's picture

commit from alex

6bf47d6 almost 2 years ago

223 Bytes

	from transformers import PretrainedConfig
	from typing import List


	class BertConfig(PretrainedConfig):
	model_type="bert"

	def __init__(
	self,
	**kwargs,
	):
	super().__init__(**kwargs)