Update README.md

README.md

@@ -188,6 +188,7 @@ Results:
 
 For more information, see: [Model Recycling](https://ibm.github.io/model-recycling/)
 
+
 # Citation
 
 More details on this [article:](https://arxiv.org/abs/2301.05948) 
@@ -200,7 +201,26 @@ More details on this [article:](https://arxiv.org/abs/2301.05948)
   year={2023}
 }
 ```
-                   
+
+# Loading a specific classifier
+
+```
+from torch import nn
+
+TASK_NAME = "hh-rlhf"
+
+class MultiTask(transformers.DebertaV2ForMultipleChoice):
+   def __init__(self, *args, **kwargs):
+        super().__init__(*args)
+        n=len(self.config.tasks)
+        cs=self.config.classifiers_size
+        self.Z = nn.Embedding(n,768)
+        self.classifiers = nn.ModuleList([torch.nn.Linear(*size) for size in cs])
+
+model = MultiTask.from_pretrained("sileod/deberta-v3-base-tasksource-nli",ignore_mismatched_sizes=True)
+task_index = {k:v for v,k in dict(enumerate(model.config.tasks)).items()}[TASK_NAME]
+model.classifier = model.classifiers[task_index] # model is ready for $TASK_NAME ! 
+```
 
 
 # Model Card Contact