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(
|