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:
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:
|
||||
|
||||
Reference in new issue
Block a user