fix linting

Signed-off-by: Sukriti-Sharma4 <sukriti.sharma4@ibm.com>
This commit is contained in:
Sukriti-Sharma4 2024-12-18 18:49:51 -07:00
parent be50c0c3b6
commit 0b2f031387

View file

@ -2691,6 +2691,7 @@ class BertModel(Model):
@Model.register("RobertaModel")
class RobertaModel(BertModel):
model_arch = gguf.MODEL_ARCH.BERT
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)