old_contents: import torch
  
  
  class BaseModel(torch.nn.Module):
      def load(self, path):
          """Load model from file.
  
          Args:
              path (str): file path
          """
          parameters = torch.load(path)
  
          if "optimizer" in parameters:
              parameters = parameters["model"]
  
          self.load_state_dict(parameters)
  
new_contents: import torch
  
  
  class BaseModel(torch.nn.Module):
      def load(self, path):
          """Load model from file.
  
          Args:
              path (str): file path
          """
          parameters = torch.load(path, map_location=torch.device('cpu'))
  
          if "optimizer" in parameters:
              parameters = parameters["model"]
  
          self.load_state_dict(parameters)
  
subject: fix error on cpu execution