Bug: Quantize stage in build shows train output

[DingmaomaoBJTU]

Description

When running the build command, the quantize stage displays train output instead of the expected quantize output.

Steps to Reproduce

Clear dataset cache.
Run modelkit build with a model that includes a quantize stage and observe the console output during the quantize step.

Expected Behavior

The quantize stage should display quantize-relevant output.

Actual Behavior

Train output is shown during the quantize stage.

Source

Reported via user feedback from Panther Lake testing.

[DingmaomaoBJTU]

Triage summary

• Component: Quantize
• Priority: P1
• Owner(s): @zhenchaoni

Customer-reported bug where the quantize stage in winml build displays train output instead of quantize output; wrong console output during a core pipeline step.