hunk.code_window: def load_checkpoint_to_cpu(path, arg_overrides=None):
       """Loads a checkpoint to CPU (with upgrading for backward compatibility)."""
       with PathManager.open(path, "rb") as f:
           state = torch.load(
               f, map_location=lambda s, l: default_restore_location(s, "cpu")
           )
  
hunk.labels: keep keep keep keep replace keep keep keep keep
hunk.after_edit:     with open(PathManager.get_local_path(path), "rb") as f:
  
hunk.file_path: fairseq/checkpoint_utils.py
hunk.type: replace
file: # Copyright (c) Facebook, Inc. and its affiliates.
  #
  # This source code is licensed under the MIT license found in the
  # LICENSE file in the root directory of this source tree.
  
  from . import BaseWrapperDataset
  
  
  class StripTokenDataset(BaseWrapperDataset):
  
      def __init__(self, dataset, id_to_strip):
          super().__init__(dataset)
          self.id_to_strip = id_to_strip
  
      def __getitem__(self, index):
          item = self.dataset[index]
          while len(item) > 0 and item[-1] == self.id_to_strip:
              item = item[:-1]
          while len(item) > 0 and item[0] == self.id_to_strip:
              item = item[1:]
          return item
  