fms-fsdp
fms-fsdp copied to clipboard
Remove `embed_*` variants of model architectures in speculator training
Assuming base model implementation supports returning embedding vectors. Contingent on https://github.com/foundation-model-stack/foundation-model-stack/pull/334