jinaai
/

jina-bert-flash-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

Markus28 commited on Mar 21

Commit

d9a681b

•

1 Parent(s): dae5c58

feat: added comment

Files changed (1) hide show

modeling_lora.py +1 -0

modeling_lora.py CHANGED Viewed

@@ -220,6 +220,7 @@ class BertLoRA(BertPreTrainedModel):
         self._register_lora(self._num_adaptions)
         self.main_params_trainable = False
         self._task_idx = None
         self.current_task = 0
     @property

         self._register_lora(self._num_adaptions)
         self.main_params_trainable = False
         self._task_idx = None
+        # By default, we select the first LoRA
         self.current_task = 0
     @property