Closed liuweie closed 1 month ago
Yes, that is correct, some portions of loading the model will also need to be changed https://github.com/McGill-NLP/llm2vec/blob/main/llm2vec/llm2vec.py#L59
Finally, if you want to define a custom template, it will require changes here. In the future versions of the library, the template will be specified outside the package (#56 )
Let me know if you have any further questions.
Yes, that is correct, some portions of loading the model will also need to be changed https://github.com/McGill-NLP/llm2vec/blob/main/llm2vec/llm2vec.py#L59
Finally, if you want to define a custom template, it will require changes here. In the future versions of the library, the template will be specified outside the package (#56 )
Let me know if you have any further questions.
thanks for your reply! By the way, I found that if I want to use my customized dadaset, I also need to add a dataset code in llm2vec package, just like E5data.py in dataset module. I think it is not a flexible way , Is there another way to do this ?
I tried the phi3 and while I managed to convert, I see the tensor names changed, e.g. the "model." prefix is missing etc...
Do you have a fork where I can see the code?
Closing as it is stale. Feel free to re-open if you have any more questions.
hi,I am trying to use llm2vec with another LLM model ( not Mistral or LLama),which parts of the code should I customize? I personally think that the code in the models section of the llm2vec library should be modified, such as adding a code similar to bidirectional_llama.py, am I right?
thx