We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f5e832f commit 6503c05Copy full SHA for 6503c05
1 file changed
src/maxdiffusion/tests/ltx2_pipeline_test.py
@@ -107,6 +107,15 @@ class MockConfig:
107
normalization_layer_epsilon = 1e-6
108
scan_layers = False
109
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"
119
120
self.text_encoder = MaxTextGemma3FeatureExtractor(
121
config=MockConfig(),
0 commit comments