Skip to content

Add flex attention flags and args to Llama on Habana#2124

Closed
kareemshaik80 wants to merge 1 commit into
huggingface:mainfrom
kareemshaik80:flex_attetnion
Closed

Add flex attention flags and args to Llama on Habana#2124
kareemshaik80 wants to merge 1 commit into
huggingface:mainfrom
kareemshaik80:flex_attetnion

Conversation

@kareemshaik80
Copy link
Copy Markdown
Contributor

What does this PR do?

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?

Signed-off-by: kareem <kshaik@habana.ai>
@regisss
Copy link
Copy Markdown
Collaborator

regisss commented Jul 9, 2025

@kareemshaik80 Can you provide a command to run so that we can test it?
Also, I suggest adding a test for it in https://github.com/huggingface/optimum-habana/blob/main/tests/test_text_generation_example.py

@astachowiczhabana
Copy link
Copy Markdown
Collaborator

This change is covered in: #2246

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.

3 participants