Skip to content

Add cross-tokenizer CSD to async distillation - #8140

Draft
kashif wants to merge 16 commits into
huggingface:mainfrom
kashif:feat/async-csd
Draft

kashif wants to merge 16 commits into
huggingface:mainfrom
kashif:feat/async-csd

Conversation

@kashif

@kashif kashif commented Nov 5, 2026 •

Copy link
Copy Markdown
Collaborator

What does this PR do?

Adds CSD to async distillation for teachers with different tokenizers, including teacher chat templates and stable chunk losses. Updates the MOPD example, docs, and cross-tokenizer tests.

Validation: focused loss/gradient and trainer tests; make precommit.

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline, Pull Request section?
  • Was this discussed/approved via a GitHub issue? Please add a link to it if that's the case.
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

AI writing disclosure

  • No AI usage: the PR was written entirely by a human.
  • AI-assisted: some parts were suggested or improved by AI, but the PR was written and reviewed by a human.
  • AI-generated: the PR was mostly or fully generated by an AI tool.

Who can review?

Anyone in the community is free to review the PR once the tests have passed.

This branch has not been deployed

No deployments
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