Update README.md

README.md

@@ -72,6 +72,10 @@ tokenized_pairs = [[tokenized_query, sent] for sent in tokenized_sentences]
 ```python
 from sentence_transformers import CrossEncoder
 model = CrossEncoder('itdainb/vietnamese-cross-encoder', max_length=256)
+
+# For fp16 usage
+model.model.half()
+
 scores = model.predict(tokenized_pairs)
 ```
 
@@ -83,6 +87,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 model = AutoModelForSequenceClassification.from_pretrained('itdainb/vietnamese-cross-encoder')
 tokenizer = AutoTokenizer.from_pretrained('itdainb/vietnamese-cross-encoder')
 
+# For fp16 usage
+model.half()
+
 features = tokenizer(tokenized_pairs, padding=True, truncation="longest_first", return_tensors="pt", max_length=256)
 
 model.eval()