Materialize non-layer modules after FSDP sharding#314

guan404ming wants to merge 2 commits into

vllm-project:mainvllm-project/speculators:mainfrom

guan404ming:fix/materialize-non-layer-fsdp-modulesguan404ming/speculators:fix/materialize-non-layer-fsdp-modulesCopy head branch name to clipboard