Skip to content

[WIP] experimental GraniteSWAForCausalLM support#1014

Draft
yannicks1 wants to merge 22 commits into
torch-spyre:mainfrom
yannicks1:swa-support
Draft

[WIP] experimental GraniteSWAForCausalLM support#1014
yannicks1 wants to merge 22 commits into
torch-spyre:mainfrom
yannicks1:swa-support

Conversation

@yannicks1

Copy link
Copy Markdown
Collaborator

Please do not review/merge. This work is very much in progress.

Description

Branch that allows loading GraniteSWA models with and without weights for testing model compilation and per-liminary bench-marking.

Note: While this branch is currently very hacky, the final PR will be light weight.

What will go away:

  • dummy weight loading code (moved out into this separate PR Feature: support --load-format dummy for random-init model loading #1013 as not specific to GraniteSWA)
  • manual vLLM model registration: ModelRegistry.register_model in platform.py (will disappear when the model is registered in vLLM (see this vllm branch here, specific commit))
  • separate gated model load with manual configuration _granite_swa_variant_from_hf (will disappear when we get actual checkpoints with model configs)

Dependencies

  • fms branch
  • hf branch
  • [vllm branch (not actually used as long as we register the model manually)]
  • special branches of lower Spyre stack...

Related Issues

#1013

Test Plan

Checklist

  • I have read the contributing guidelines
  • My code follows the project's code style (run bash format.sh)
  • I have added tests for my changes (if applicable)
  • I have updated the documentation (if applicable)
  • My commits include a Signed-off-by: line (DCO compliance)

yannicks1 added 19 commits May 15, 2026 16:59
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
@github-actions

Copy link
Copy Markdown

👋 Hi! Thank you for contributing.
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, run ./format.sh.
Now you are good to go 🚀.

We also recommend installing prek and configuring it to check your code before every local commit.

Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
Signed-off-by: Yannick Schnider <Yannick.Schnider1@ibm.com>
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.

1 participant