From 0a0e1884f7eb692234d4ec9965bc53275c58e7b7 Mon Sep 17 00:00:00 2001 From: Canlin Guo Date: Thu, 24 Sep 2026 21:34:00 +0800 Subject: [PATCH] [CI/Build] Initialize RoPE frequencies in MiniMax H3 TeaCache tests (#8125) Signed-off-by: Canlin Guo --- tests/diffusion/cache/test_teacache_extractors.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/diffusion/cache/test_teacache_extractors.py b/tests/diffusion/cache/test_teacache_extractors.py index 7dd613f5f..cdaef2c7c 100644 --- a/tests/diffusion/cache/test_teacache_extractors.py +++ b/tests/diffusion/cache/test_teacache_extractors.py @@ -652,6 +652,9 @@ class TestMiniMaxH3Extractor(BaseExtractorTest): monkeypatch.setattr(h3, "get_tensor_model_parallel_world_size", lambda: 1) model = h3.MiniMaxH3DiTModel(_minimax_h3_small_od_config(), quant_config=None) + # Normally loaded from the checkpoint; torch.empty() can contain NaNs. + # With one frequency per axis, the RoPE inverse frequency is 1. + model.rope.inv_freq.fill_(1.0) for submodule in model.modules(): if isinstance(submodule, h3.MiniMaxH3Attention): submodule.rope._forward_method = submodule.rope.forward_native