search

xbmxb / RAG-query-rewriting

122 stars 8 forks source link

open source trained model #7

Open cooper12121 opened 3 months ago

cooper12121 commented 3 months ago

Excuse me , i want to reproduce your work, can you offer the trained model?, i want to predict the results.

catyung commented 2 months ago

@cooper12121 Hi, I think you should be able to get the model from Google Drive, mentioned in the readme : https://github.com/xbmxb/RAG-query-rewriting/tree/main/result_models

However, I personally took 2 hours to download the models from Google Drive

So, I have complied checkpoint_20, released by @xbmxb on Google Drive, into a Transformers compatible format, I think it will be easier for the community to use the model.

Here is the model & I also add a inference code with a rewrite prompt, refer to the research paper : https://huggingface.co/catyung/t5l-turbo-hotpot-0331

Thanks for @xbmxb releasing the code, model and research <3