Skip to content

Conversation

yannicks1
Copy link
Collaborator

Supporting granite 2b & 3b: work in progress

consumes this fms branch: https://github.com/foundation-model-stack/foundation-model-stack/tree/granite-2b-expand-q-k-128

Copy link

github-actions bot commented Oct 1, 2025

👋 Hi! Thank you for contributing to vLLM support on Spyre.
Just a reminder: Make sure that your code passes all the linting checks, otherwise your PR won't be able to be merged. To do so, first install the linting requirements, then run format.sh and commit the changes. This can be done with uv directly:

uv sync --frozen --group lint --active --inexact

Or this can be done with pip:

uv pip compile --group lint > requirements-lint.txt
pip install -r requirements-lint.txt
bash format.sh

Now you are good to go 🚀

Signed-off-by: Yannick Schnider <[email protected]>
Signed-off-by: Yannick Schnider <[email protected]>
self.kv_cache_specs['head_dim'] = self.config.hidden_size // \
self.config.num_attention_heads

# *** ALERT *** Granite 2b hack for AIU Compiler
Copy link
Collaborator

Choose a reason for hiding this comment

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

Do we know if this is a temporary hack? Then perhaps we can add a compatibility test

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

will see how they implement it properly in fms. I do just mirror their strategy here... (very much work in progress)

Signed-off-by: Yannick Schnider <[email protected]>
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