[dist][moe] fix add moe_context for big models#2405

kylesayrs merged 2 commits intomainvllm-project/llm-compressor:mainfrom

big_model_ddp_fixvllm-project/llm-compressor:big_model_ddp_fixCopy head branch name to clipboard