Skip to content

Commit d2c5cad

Browse files
committed
fix(mmgen): align trajectory_timesteps type annotations
1 parent eebe663 commit d2c5cad

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/sglang/multimodal_gen/runtime/pipelines_core/schedule_batch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ class Req:
133133
# Component modules (populated by the pipeline)
134134
modules: dict[str, Any] = field(default_factory=dict)
135135

136-
trajectory_timesteps: list[torch.Tensor] | None = None
136+
trajectory_timesteps: torch.Tensor | None = None
137137
trajectory_latents: torch.Tensor | None = None
138138
trajectory_log_probs: torch.Tensor | None = None
139139
trajectory_audio_latents: torch.Tensor | None = None
@@ -332,7 +332,7 @@ class OutputBatch:
332332
output: torch.Tensor | None = None
333333
audio: torch.Tensor | None = None
334334
audio_sample_rate: int | None = None
335-
trajectory_timesteps: list[torch.Tensor] | None = None
335+
trajectory_timesteps: torch.Tensor | None = None
336336
trajectory_latents: torch.Tensor | None = None
337337
trajectory_log_probs: torch.Tensor | None = None
338338
trajectory_decoded: list[torch.Tensor] | None = None

0 commit comments

Comments
 (0)