Skip to content

Commit fc823d2

Browse files
committed
unit test
1 parent f8557e0 commit fc823d2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/maxdiffusion/tests/wan_transformer_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,7 @@ def test_quantize_transformer_enabled(self, mock_get_dummy_inputs, mock_quantize
396396
mock_config.weight_quantization_calibration_method = "fixed,-224,224"
397397
mock_config.act_quantization_calibration_method = "fixed,-224,224"
398398
mock_config.bwd_quantization_calibration_method = "absmax"
399-
mock_config.global_batch_size_to_train_on=32
399+
mock_config.global_batch_size_to_train_on = 32
400400

401401
mock_model = Mock(spec=WanModel)
402402
mock_pipeline = Mock()

0 commit comments

Comments
 (0)