feat: add current_task to forward

modeling_lora.py

@@ -259,7 +259,9 @@ class BertLoRA(BertPreTrainedModel):
             partial(LoRAParametrization.select_task_for_layer, task_idx=task_idx)
         )
 
-    def forward(self, *args, **kwargs):
+    def forward(self, *args, **kwargs, current_task: Union[None, int] = -1):
+        if current_task is None or current_task >= 0:
+            self.current_task = current_task
         return self.bert(*args, **kwargs)
 
     def parameters(self, recurse: bool = True) -> Iterator[Parameter]: