Skip to content

Add open-source model support & random multi-needle insertion#42

Open
gauss5930 wants to merge 2 commits intogkamradt:mainfrom
gauss5930:open-models
Open

Add open-source model support & random multi-needle insertion#42
gauss5930 wants to merge 2 commits intogkamradt:mainfrom
gauss5930:open-models

Conversation

@gauss5930
Copy link
Copy Markdown

I have enabled NIAH to be performed on open-source models existing in the HuggingFace Hub. In addition, I introduce a method to insert needles more randomly and evenly in the case of Multi-needle.

Please check the update of README for more specific information!

@saharmor
Copy link
Copy Markdown

AI21 just released a new model on Hugging Face that supports a 256k context window. This PR will enable testing it. Thanks @gauss5930.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants