jupyterjazz
commited on
Commit
•
c380b5a
1
Parent(s):
5ed05aa
Update modeling_lora.py
Browse files- 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.
|
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):
|