Spaces:
Running
Running
| 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. | |