removed comments from conversion script

Author: Serenagu525

src/maxdiffusion/checkpointing/checkpointing_utils.py

@@ -213,13 +213,11 @@ def load_state_if_possible(
     max_logging.log(f"restoring from this run's directory latest step {latest_step}")
     try:
       if not enable_single_replica_ckpt_restoring:
-        # item = {checkpoint_item: orbax.checkpoint.args.PyTreeRestore(item=abstract_unboxed_pre_state)}
-        # return checkpoint_manager.restore(latest_step, args=orbax.checkpoint.args.Composite(**item))  #currently changed to this
         if checkpoint_item == " ":
           return checkpoint_manager.restore(latest_step, args=ocp.args.StandardRestore(abstract_unboxed_pre_state))
         else:
           item = {checkpoint_item: orbax.checkpoint.args.PyTreeRestore(item=abstract_unboxed_pre_state)}
-          return checkpoint_manager.restore(latest_step, args=orbax.checkpoint.args.Composite(**item))  #currently changed to this
+          return checkpoint_manager.restore(latest_step, args=orbax.checkpoint.args.Composite(**item)) 
 
       def map_to_pspec(data):
         pspec = data.sharding.spec

src/maxdiffusion/generate_ltx_video.py

@@ -50,7 +50,7 @@ def run(config):
 
 
   transformer = Transformer3DModel(**model_config, dtype=jnp.float32, gradient_checkpointing="matmul_without_batch", sharding_mesh=mesh)
-  transformer_param_shapes = transformer.init_weights(in_channels, model_config['caption_channels'], eval_only = True) #use this to test!
+  transformer_param_shapes = transformer.init_weights(in_channels, model_config['caption_channels'], eval_only = True)
 
   weights_init_fn = functools.partial(
       transformer.init_weights, 

src/maxdiffusion/max_utils.py

@@ -402,7 +402,6 @@ def setup_initial_state(
           config.enable_single_replica_ckpt_restoring,
       )
       if state:
-        ###!Edited
         if checkpoint_item == " ":
           state = state
         else: