Skip to content

[Update] new functions and bug fixes#90

Open
yeahjack wants to merge 6 commits intoRUC-NLPIR:mainfrom
yeahjack:refiner_multi_gpu
Open

[Update] new functions and bug fixes#90
yeahjack wants to merge 6 commits intoRUC-NLPIR:mainfrom
yeahjack:refiner_multi_gpu

Conversation

@yeahjack
Copy link
Contributor

@yeahjack yeahjack commented Oct 6, 2024

During this PR, I made the following changes:

  1. Add support for flash-attention-2 in HFCausalLMGenerator, and add Llama-3 special token when initializing the model.
  2. Add refiner on multi GPU, hence accelerating the refining process by a lot.
  3. Fix a bug for selective-context for list out of bounds.
  4. Add paths for models in recomp and fix loading.
  5. Fix a bug for intermediate_data.json saving when using bm25s.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments