sgugger
/

my-bert-model

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

my-bert-model / modeling.py

sgugger's picture

Fix syntax error

1ddc92c almost 4 years ago

179 Bytes

	from transformers import BertModel

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