Skip to content

Commit

Permalink
Do not change existing lines
Browse files Browse the repository at this point in the history
  • Loading branch information
NielsRogge committed Feb 8, 2025
1 parent 2d5c535 commit 4259bf4
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions verl/workers/fsdp_workers.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@
from verl.utils.checkpoint.fsdp_checkpoint_manager import FSDPCheckpointManager
from verl.workers.sharding_manager.fsdp_ulysses import FSDPUlyssesShardingManager


from codetiming import Timer

logger = logging.getLogger(__file__)
Expand Down Expand Up @@ -545,7 +544,7 @@ def compute_ref_log_prob(self, data: DataProto):

torch.cuda.empty_cache()
return output

@register(dispatch_mode=Dispatch.ONE_TO_ALL)
def save_checkpoint(self, local_path, hdfs_path=None, global_step=0):
# only support save and load ckpt for actor
Expand Down

0 comments on commit 4259bf4

Please sign in to comment.