fix: fixed from_bert method
Browse files- modeling_lora.py +2 -2
modeling_lora.py
CHANGED
@@ -225,10 +225,10 @@ class BertLoRA(BertPreTrainedModel):
|
|
225 |
param.requires_grad_(val)
|
226 |
|
227 |
@classmethod
|
228 |
-
def from_bert(cls, *args,
|
229 |
bert = BertModel.from_pretrained(*args, **kwargs)
|
230 |
config = JinaBertConfig.from_pretrained(*args, **kwargs)
|
231 |
-
return cls(config, bert=bert
|
232 |
|
233 |
@classmethod
|
234 |
def from_pretrained(
|
|
|
225 |
param.requires_grad_(val)
|
226 |
|
227 |
@classmethod
|
228 |
+
def from_bert(cls, *args, **kwargs):
|
229 |
bert = BertModel.from_pretrained(*args, **kwargs)
|
230 |
config = JinaBertConfig.from_pretrained(*args, **kwargs)
|
231 |
+
return cls(config, bert=bert)
|
232 |
|
233 |
@classmethod
|
234 |
def from_pretrained(
|