jupyterjazz commited on
Commit
c380b5a
1 Parent(s): 5ed05aa

Update modeling_lora.py

Browse files
Files changed (1) hide show
  1. modeling_lora.py +1 -1
modeling_lora.py CHANGED
@@ -311,7 +311,7 @@ class XLMRobertaLoRA(XLMRobertaPreTrainedModel):
311
  def forward(self, *args, current_task: Union[None, int] = -1, **kwargs):
312
  if current_task is None or current_task >= 0:
313
  self.current_task = current_task
314
- return self.bert(*args, **kwargs)
315
 
316
  def parameters(self, recurse: bool = True) -> Iterator[Parameter]:
317
  for _, param in self.named_parameters(recurse=recurse):
 
311
  def forward(self, *args, current_task: Union[None, int] = -1, **kwargs):
312
  if current_task is None or current_task >= 0:
313
  self.current_task = current_task
314
+ return self.roberta(*args, **kwargs)
315
 
316
  def parameters(self, recurse: bool = True) -> Iterator[Parameter]:
317
  for _, param in self.named_parameters(recurse=recurse):