Update README.md

README.md

@@ -21,7 +21,9 @@ You can use this model simply as a re-ranker, note now the model is only availab
 from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("howard-hou/RankingPrompterForPreTraining-small")
-model = AutoModel.from_pretrained("howard-hou/RankingPrompterForPreTraining-small")
+# trust_remote_code=True 很重要，否则不会读取到正确的模型
+model = AutoModel.from_pretrained("howard-hou/RankingPrompterForPreTraining-small",
+                                  trust_remote_code=True)
 
 #
 documents = [