megatron_patch/model/qwen2_moe/__init__.py (1 lines of code) (raw):