Markus28 commited on
Commit
151f328
·
1 Parent(s): 20706dd

fix: fixed from_bert method

Browse files
Files changed (1) hide show
  1. 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, num_adaptions=1, **kwargs):
229
  bert = BertModel.from_pretrained(*args, **kwargs)
230
  config = JinaBertConfig.from_pretrained(*args, **kwargs)
231
- return cls(config, bert=bert, num_adaptions=num_adaptions)
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(