feat: use property in LoRA parametrization

modeling_lora.py

@@ -116,8 +116,13 @@ class LoRAParametrization(nn.Module):
     def forward(self, X):
         return self.forward_fn(X)
 
-    def select_task(self, task=None):
-        self.current_task = task
+    @property
+    def current_task(self):
+        return self._current_task
+
+    @current_task.setter
+    def current_task(self, task: Union[None, int]):
+        self._current_task = task
         if task is None:
             self.forward_fn = lambda x: x
         else:
@@ -192,7 +197,7 @@ class LoRAParametrization(nn.Module):
     @classmethod
     def select_task_for_layer(cls, layer: nn.Module, task_idx: Optional[int] = None):
         if isinstance(layer, LoRAParametrization):
-            layer.select_task(task_idx)
+            layer.current_task = task_idx
 
 
 class BertLoRA(BertPreTrainedModel):