update vllm.yml after rebasing

Author: A9isha

src/maxtext/configs/inference/vllm.yml

@@ -70,6 +70,8 @@ logical_axis_rules: [
                       ['kv', []],
                       ['embed', ['expert', 'attn_dp_expert']],
                       ['embed', ['attn_dp_expert']],
+                      ['embed_moe', ['expert', 'attn_dp_expert']],
+                      ['embed_moe', ['attn_dp_expert']],
                       ['embed_tensor_transpose', ['attn_dp', 'model']],
                       ['embed_no_exp', []],
                       ['embed_no_exp_moe', []],