Update README.md

README.md

@@ -8,6 +8,10 @@ tags:
 ---
 This version of DNABERT2 has been changed to be able to output the attention too, for attention analysis. 
 
+**To the author of DNABERT2, feel free to use those modifications.**
+
+Use  ```--model_name_or_path jaandoui/DNABERT2_AttentionExtracted``` instead of the original repository to have access to the attention.
+
 Most of the modifications were done in Bert_Layer.py. 
 It has been modified especially for fine tuning and hasn't been tried for pretraining.
 Before or next to each modification, you can find "JAANDOUI" so to see al modifications, search for "JAANDOUI".
@@ -19,8 +23,6 @@ activate the extraction of attention: ```output_attentions=True``` (and ```retur
 You can now extract the attention in ```outputs.attentions```
 Read more about model outputs here: https://huggingface.co/docs/transformers/v4.40.2/en/main_classes/output#transformers.utils.ModelOutput
 
-To the author of DNABERT2, feel free to use those modifications.
-
 The official link to DNABERT2 [DNABERT-2: Efficient Foundation Model and Benchmark For Multi-Species Genome
 ](https://arxiv.org/pdf/2306.15006.pdf).