container/sagemaker-pytorch-inferentia-serving/src/sagemaker_pytorch_inferentia_serving_container/default_inference_handler.py [32:40]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        model_files = []
        for f in os.listdir(model_dir):
            if os.path.isfile(f):
                name, ext = os.path.splitext(f)
                if ext == ".pt" or ext == ".pth":
                    model_files.append(f)
        if len(model_files) != 1:
            raise ValueError("Exactly one .pth or .pt file is required for PyTorch models: {}".format(model_files))
        return torch.jit.load(model_files[0])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



container/sample-inf1/inf1_pt.py [28:36]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  model_files = []
  for f in os.listdir(model_dir):
      if os.path.isfile(f):
          name, ext = os.path.splitext(f)
          if ext == ".pt" or ext == ".pth":
              model_files.append(f)
  if len(model_files) != 1:
      raise ValueError("Exactly one .pth or .pt file is required for PyTorch models: {}".format(model_files))
  return torch.jit.load(model_files[0])
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -