remove prints

Author: pradeepfn

src/forge/actors/policy.py

@@ -413,11 +413,6 @@ async def _load_tensor_parallel_state_dict(
         # setting explictly to llama3 for now as its our only use case
         sharding = VLLMSharding(self.tensor_parallel_size, self.rank)
 
-        # key_str = ""
-        # for param_name in current_state_dict.keys():
-        #    key_str += f" {self.state_dict_key}/{version}/{param_name}\n"
-        # logger.warning(f"############### policy get keys : {key_str}")
-
         for param_name in current_state_dict.keys():
             current_tensor = current_state_dict[param_name]
             # Load the full tensor from torchstore

src/forge/actors/trainer.py

@@ -304,12 +304,6 @@ async def push_weights(self) -> None:
                 [gate, up], dim=0
             )
 
-        # Remove before landing
-        # key_str = ""
-        # for key, _ in hf_state_dict.items():
-        #    key_str += f" model_state_dict/{self.current_step}/{key}\n"
-        # logger.warning(f"rltrainer, put_state_dict keys : {key_str}")
-
         await ts.put_state_dict(
             state_dict=hf_state_dict,
             key=f"model_state_dict/{self.current_step}",