Skip to content

sarkar/Fix mistral after syn1.15 update#858

Merged
regisss merged 1 commit into
mainfrom
sarkar/fix_mistral_allocatekv
Apr 3, 2024
Merged

sarkar/Fix mistral after syn1.15 update#858
regisss merged 1 commit into
mainfrom
sarkar/fix_mistral_allocatekv

Conversation

@ssarkar2
Copy link
Copy Markdown
Contributor

@ssarkar2 ssarkar2 commented Apr 2, 2024

What does this PR do?

After this commit: #831
allocate_kv_cache has 1 less argument

This was fixed for llama but not for mistral: https://github.com/huggingface/optimum-habana/blob/main/optimum/habana/transformers/models/llama/modeling_llama.py#L823

Fixes # (issue)

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

@ssarkar2 ssarkar2 requested a review from regisss as a code owner April 2, 2024 21:01
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Copy link
Copy Markdown
Collaborator

@libinta libinta left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ci update will be in different PR.

@regisss regisss merged commit b54cb8a into main Apr 3, 2024
@regisss regisss deleted the sarkar/fix_mistral_allocatekv branch April 3, 2024 10:32
regisss pushed a commit that referenced this pull request Apr 3, 2024
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