[AMD] [CI] Enable 3 nested unit tests needing harness stub fixes (#31843)
Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
@@ -1091,6 +1091,11 @@ _AMD_READY_NESTED_UNIT_TESTS = (
|
||||
"sana_wm/test_streaming_cached.py",
|
||||
"sana_wm/test_streaming_stage.py",
|
||||
"sana_wm/test_streaming_vae.py",
|
||||
# Enabled with small test-harness stub fixes (see this PR's test edits).
|
||||
"progressive_resolution/test_progressive.py",
|
||||
"sana_wm/test_streaming_realtime_path.py",
|
||||
# Stub gap already fixed upstream; only needs enabling here.
|
||||
"realtime/test_lingbot_causal_denoising.py",
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -9,6 +9,7 @@ from types import SimpleNamespace
|
||||
import torch
|
||||
|
||||
from sglang.multimodal_gen.configs.sample.sampling_params import SamplingParams
|
||||
from sglang.multimodal_gen.runtime.distributed.cfg_policy import CFGPolicy
|
||||
from sglang.multimodal_gen.runtime.pipelines_core.stages.progressive_resolution.denoising import (
|
||||
ProgressiveDenoisingStage,
|
||||
ProgressiveDenoisingStageRouter,
|
||||
@@ -500,6 +501,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
||||
ctx = SimpleNamespace(
|
||||
latents=torch.zeros(B, new_num_img, self._IN_C),
|
||||
extra=self._make_ctx_extra(B, old_num_img, max_text_tokens),
|
||||
# Non-None so _on_resolution_change proceeds (it early-returns when
|
||||
# cfg_policy is None); the value itself is not used further here.
|
||||
cfg_policy=CFGPolicy(),
|
||||
)
|
||||
batch = SimpleNamespace(
|
||||
extra={
|
||||
@@ -555,6 +559,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
||||
ctx = SimpleNamespace(
|
||||
latents=torch.zeros(B, new_grid_h * new_grid_w, self._IN_C),
|
||||
extra=self._make_ctx_extra(B, old_grid_h * old_grid_w, max_text_tokens),
|
||||
# Non-None so _on_resolution_change proceeds (it early-returns when
|
||||
# cfg_policy is None); the value itself is not used further here.
|
||||
cfg_policy=CFGPolicy(),
|
||||
)
|
||||
batch = SimpleNamespace(extra={"ideogram4": old_data})
|
||||
|
||||
@@ -594,6 +601,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
||||
ctx = SimpleNamespace(
|
||||
latents=torch.zeros(B, grid_h * grid_w, self._IN_C),
|
||||
extra=self._make_ctx_extra(B, old_num_img, max_text_tokens),
|
||||
# Non-None so _on_resolution_change proceeds (it early-returns when
|
||||
# cfg_policy is None); the value itself is not used further here.
|
||||
cfg_policy=CFGPolicy(),
|
||||
)
|
||||
batch = SimpleNamespace(
|
||||
extra={
|
||||
|
||||
@@ -47,6 +47,7 @@ def _global_args():
|
||||
comfyui_mode=False,
|
||||
enable_cfg_parallel=False,
|
||||
enable_torch_compile=False,
|
||||
enable_breakable_cuda_graph=False,
|
||||
attention_backend=None,
|
||||
# DenoisingStage.__init__ reads this for its CFG-parallel plumbing.
|
||||
pipeline_config=SimpleNamespace(
|
||||
|
||||
Reference in New Issue
Block a user