Revert "[Bugfix][MiniMax-H3] Keep keyframe VAE encoding rank-local"

This reverts commit 7a2a16f8af.

Signed-off-by: david6666666 <530634352@qq.com>
This commit is contained in:
david6666666 committed 2026-09-17 09:20:18 +00:00
1 parent 00b2eb9399
commit 52ec7e4d61
2 files changed
+3 -9

No files matched your search

@@ -3,7 +3,7 @@
import json
import sys
from contextlib import contextmanager, nullcontext
from contextlib import contextmanager
from multiprocessing.reduction import ForkingPickler
from types import SimpleNamespace
from typing import Any, TypeVar
@@ -2821,7 +2821,6 @@ def test_keyframe_encode_enters_backend_context(monkeypatch, fail_encode):
assert events == ["enter"]
assert image is input_image
assert use_fp16_latent
assert not self.parallel_tiling
events.append("encode")
if fail_encode:
raise RuntimeError("encode failed")
@@ -2835,8 +2834,6 @@ def test_keyframe_encode_enters_backend_context(monkeypatch, fail_encode):
"latents_mean": [0.0],
"latents_std": [1.0],
}
video_vae.parallel_size = 4
monkeypatch.setattr(video_vae, "_encoder_tiling_context", lambda _height, _width: nullcontext())
monkeypatch.setattr(vae_module, "_minimax_h3_keyframe_encode_context", track_context)
if fail_encode: