Skip to content

Yulin mistral#26

Open
yulinw2 wants to merge 28 commits intoInfini-AI-Lab:mainfrom
yulinw2:yulin_mistral
Open

Yulin mistral#26
yulinw2 wants to merge 28 commits intoInfini-AI-Lab:mainfrom
yulinw2:yulin_mistral

Conversation

@yulinw2
Copy link

@yulinw2 yulinw2 commented Feb 21, 2025

Add Mistral Model based on main branch.

  1. Supporting Sliding window by introducing a sliding KV cache (Can test it in generate.py by using sliding_window and sliding_size)
  2. Supporting GQA (use flashinfer or self impelemented gqa function in cache.py, both can work)

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.

4 participants