comments removed

Serenagu525 · Serenagu525 · commit d1c304da1a67 · 2025-06-27T00:20:19.000Z
diff --git a/src/maxdiffusion/generate_ltx_video.py b/src/maxdiffusion/generate_ltx_video.py
@@ -77,8 +77,7 @@ def run(config):
   )
   transformer_param_shapes = transformer.init_weights(  # noqa: F841
       in_channels, key, model_config["caption_channels"], eval_only=True
-  )  # use this to test!
-
+  )  
   weights_init_fn = functools.partial(
       transformer.init_weights, in_channels, key, model_config["caption_channels"], eval_only=True
   )
@@ -115,7 +114,4 @@ def main(argv: Sequence[str]) -> None:
   app.run(main)
 
 
-###setup_initial_state, can optionally load from checkpoint
-
 
-# end to end steps from ltx repo: pipeline_ltx_video.py

Original file line number	Diff line number	Diff line change
`@@ -77,8 +77,7 @@ def run(config):`
`77`	`77`	`)`
`78`	`78`	`transformer_param_shapes = transformer.init_weights( # noqa: F841`
`79`	`79`	`in_channels, key, model_config["caption_channels"], eval_only=True`
`80`		`- ) # use this to test!`
`81`		`-`
	`80`	`+ )`
`82`	`81`	`weights_init_fn = functools.partial(`
`83`	`82`	`transformer.init_weights, in_channels, key, model_config["caption_channels"], eval_only=True`
`84`	`83`	`)`
`@@ -115,7 +114,4 @@ def main(argv: Sequence[str]) -> None:`
`115`	`114`	`app.run(main)`
`116`	`115`
`117`	`116`
`118`		`-###setup_initial_state, can optionally load from checkpoint`
`119`		`-`
`120`	`117`
`121`		`-# end to end steps from ltx repo: pipeline_ltx_video.py`