load transformer error

Author: Serenagu525

src/maxdiffusion/models/ltx_video/transformers/transformer3d.py

@@ -154,6 +154,7 @@ def scale_shift_table_init(key):
                 matmul_precision=self.matmul_precision,
             )
     def init_weights(self, key, in_channels, caption_channels, eval_only=True):
+        import pdb; pdb.set_trace()
         example_inputs = {}
         batch_size, num_tokens = 4, 256
         input_shapes = {

src/maxdiffusion/pipelines/ltx_video/ltx_video_pipeline.py

@@ -240,7 +240,7 @@ def load_transformer(cls, config):
             tx=None,
             config=config,
             mesh=mesh,
-            weights_init_fn=weights_init_fn,
+            weights_init_fn=None,
             checkpoint_manager=checkpoint_manager,
             checkpoint_item=" ",
             model_params=None,