Skip to content

Commit 3668dde

Browse files
committed
fix unit test
1 parent 15b61a8 commit 3668dde

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/maxdiffusion/tests/wan_transformer_test.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -383,7 +383,9 @@ def test_quantize_transformer_enabled(self, mock_get_dummy_inputs, mock_quantize
383383
mock_config.quantization = "fp8_full"
384384
mock_config.qwix_module_path = ".*"
385385
mock_config.per_device_batch_size = 1
386-
mock_config.quantization_calibration_method = "absmax"
386+
mock_config.weight_quantization_calibration_method = "fixed,-224,224"
387+
mock_config.act_quantization_calibration_method = "fixed,-224,224"
388+
mock_config.bwd_quantization_calibration_method = "absmax"
387389

388390
mock_model = Mock(spec=WanModel)
389391
mock_pipeline = Mock()

0 commit comments

Comments
 (0)