ERROR loading model: Failed to load MHR model from /Users/Jake.Reardon@MLSE.com/.cache/huggingface/hub/models--facebook--sam-3d-body-vith/snapshots/4e4de91d0f1d0ca64ea8072d52fdbac8b8c905e9/assets/mhr_model.pt: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch. 
Fallback error: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch. 
Traceback (most recent call last):
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/heads/mhr_head.py", line 117, in __init__
    self.mhr = torch.jit.load(
               ^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/.venv/lib/python3.11/site-packages/torch/jit/_serialization.py", line 162, in load
    cpp_module = torch._C.import_ir_module(cu, str(f), map_location, _extra_files, _restore_shapes)  # type: ignore[call-arg]
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 133, in patched_import_ir
    return original_import_ir(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 133, in patched_import_ir
    return original_import_ir(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch. 

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/heads/mhr_head.py", line 125, in __init__
    self.mhr = torch.jit.load(mhr_model_path, map_location=cpu_map)
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/.venv/lib/python3.11/site-packages/torch/jit/_serialization.py", line 156, in load
    map_location = validate_map_location(map_location)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/.venv/lib/python3.11/site-packages/torch/jit/_serialization.py", line 176, in validate_map_location
    raise ValueError(
ValueError: map_location should be either None, string or torch.device, but got type: <class 'function'>

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/heads/mhr_head.py", line 129, in __init__
    self.mhr = torch.jit.load(mhr_model_path, map_location=torch.device('cpu'))
               ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 70, in patched_jit_load
    return original_jit_load(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/.venv/lib/python3.11/site-packages/torch/jit/_serialization.py", line 162, in load
    cpp_module = torch._C.import_ir_module(cu, str(f), map_location, _extra_files, _restore_shapes)  # type: ignore[call-arg]
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 133, in patched_import_ir
    return original_import_ir(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/torch_uint32_patch.py", line 133, in patched_import_ir
    return original_import_ir(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch. 

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/app/helper/model_loader.py", line 70, in load_model
    model, model_cfg = load_sam_3d_body_hf('facebook/sam-3d-body-vith', use_auth_token=hf_token)
                       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/build_models.py", line 51, in load_sam_3d_body_hf
    return load_sam_3d_body(checkpoint_path=ckpt_path, mhr_path=mhr_path)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/build_models.py", line 29, in load_sam_3d_body
    model = SAM3DBody(model_cfg)
            ^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/meta_arch/base_model.py", line 25, in __init__
    self._initialze_model(**kwargs)
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/meta_arch/sam3d_body.py", line 58, in _initialze_model
    self.head_pose = build_head(self.cfg, self.cfg.MODEL.PERSON_HEAD.POSE_TYPE)
                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/heads/__init__.py", line 10, in build_head
    return MHRHead(
           ^^^^^^^^
  File "/Users/Jake.Reardon@MLSE.com/mlse-player-3d/sam-3d-body/sam_3d_body/models/heads/mhr_head.py", line 132, in __init__
    raise RuntimeError(
RuntimeError: Failed to load MHR model from /Users/Jake.Reardon@MLSE.com/.cache/huggingface/hub/models--facebook--sam-3d-body-vith/snapshots/4e4de91d0f1d0ca64ea8072d52fdbac8b8c905e9/assets/mhr_model.pt: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch. 
Fallback error: THPDevice_Check(map_location.ptr()) INTERNAL ASSERT FAILED at "/Users/runner/work/pytorch/pytorch/pytorch/torch/csrc/jit/python/script_init.cpp":1851, please report a bug to PyTorch.