Update README.md

README.md

@@ -46,6 +46,8 @@ There are 2 versions of the pre-trained model,
 * v1 - Random Masking: We randomly choose tokens to mask in the MLM task;
 * v2 - Selective Masking: As we want the model to learn more about the financial context, we selectively choose tokens to mask in the MLM task. We rely on the Loughran-McDonald dictionary to choose the important tokens to masked.
 
+The argument `version` in the method `load_pretrained` of the `LongBERTModel` allows one to choose which version is loaded.
+
 # Instruction to load the pre-trained model
 * Clone the git repo
 ```
@@ -69,7 +71,7 @@ backbone = 'minhtriphan/LongFinBERT-base'
 
 tokenizer = LongBERTTokenizer.from_pretrained(backbone)
 config = LongBERTConfig.from_pretrained(backbone)
-model = LongBERTModel.from_pretrained(backbone)
+model = LongBERTModel.from_pretrained(backbone, version = 'v1')
 ```
 
 # Model usage