Support mbridge distillation for any_model#904

Open

danielkorzekwa wants to merge 6 commits intodkorzekwa/any_modelfrom

dkorzekwa/any_model_mbridge_distillation

Commits on Feb 18, 2026

Add MBridge distillation support for AnyModel checkpoints
danielkorzekwa
committed
Add missing files from modelopt/torch/puzzletron/export/mbridge
danielkorzekwa
committed

Commits on Feb 19, 2026

Commits on Feb 20, 2026

Fixing distillation for heterogenous models (call self.teacher.finalize() on DistillationProvider.provide()
danielkorzekwa
committed