Fix diffusion API initialization (#7868)

Signed-off-by: Tianyu Guo <guoty9@mail2.sysu.edu.cn>
This commit is contained in:
Tianyu Guo authored and GitHub committed 2026-09-20 12:08:50 +08:00
1 parent 9dabaf8b80
commit 3ea9e60501
4 files changed
+62 -11

No files matched your search

@@ -36,6 +36,21 @@ def _roundtrip_diffusion_config(**kwargs) -> OmniDiffusionConfig:
return OmniDiffusionConfig.from_kwargs(**engine_args)
def test_direct_config_extras_are_independent_dicts():
first = OmniDiffusionConfig(model="test-model")
second = OmniDiffusionConfig(model="test-model")
assert first.extras.get("prefix_kv_cache_dtype") is None
first.extras["prefix_kv_cache_dtype"] = "fp8"
assert second.extras == {}
def test_direct_config_preserves_explicit_extras():
config = OmniDiffusionConfig(model="test-model", extras={"prefix_kv_cache_dtype": "fp8"})
assert config.extras["prefix_kv_cache_dtype"] == "fp8"
class TestParallelConfigPropagation:
"""Core regression tests: parallel_config must survive serialization."""