sgugger
/

my-bert-model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions Community

my-bert-model / modeling.py

sgugger's picture

add model

0ca266b almost 4 years ago

177 Bytes

	from transformers import BertModel

	class MyBertModel(BertModel):
	def forward(self, args, *kwargs):
	print("Success")
	return super.forward(args, *kwargs)