Skip to content

Commit 6503c05

Browse files
committed
fix
1 parent f5e832f commit 6503c05

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

src/maxdiffusion/tests/ltx2_pipeline_test.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,15 @@ class MockConfig:
107107
normalization_layer_epsilon = 1e-6
108108
scan_layers = False
109109
param_scan_axis = 1
110+
max_prefill_predict_length = 512
111+
per_device_batch_size = 1
112+
max_target_length = 512
113+
rope_min_timescale = 1
114+
rope_max_timescale = 10000
115+
rope_type = "interleaved"
116+
rope_embedding_dims = 16
117+
rope_use_scale = False
118+
shard_mode = "auto"
110119

111120
self.text_encoder = MaxTextGemma3FeatureExtractor(
112121
config=MockConfig(),

0 commit comments

Comments
 (0)