runtime error

Exit code: 1. Reason: | 1/4 [00:02<00:08, 2.85s/it] Loading checkpoint shards: 50%|█████ | 2/4 [00:06<00:06, 3.05s/it] Loading checkpoint shards: 75%|███████▌ | 3/4 [00:09<00:03, 3.12s/it] Loading checkpoint shards: 100%|██████████| 4/4 [00:09<00:00, 2.47s/it] generation_config.json: 0%| | 0.00/184 [00:00<?, ?B/s] generation_config.json: 100%|██████████| 184/184 [00:00<00:00, 1.23MB/s] config.json: 0%| | 0.00/546 [00:00<?, ?B/s] config.json: 100%|██████████| 546/546 [00:00<00:00, 5.59MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 25, in <module> mast3r = MASt3R(device=DEVICE) File "/home/user/app/mast3r.py", line 82, in __init__ self.model = AsymmetricMASt3R.from_pretrained(model_id).to(self.device) File "/home/user/app/mast3r.py", line 29, in from_pretrained return super(AsymmetricMASt3R, cls).from_pretrained(pretrained_model_name_or_path, **kw) File "/usr/local/lib/python3.10/site-packages/mini_dust3r/model.py", line 98, in from_pretrained return super(AsymmetricCroCo3DStereo, cls).from_pretrained( File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/huggingface_hub/hub_mixin.py", line 550, in from_pretrained instance = cls._from_pretrained( File "/usr/local/lib/python3.10/site-packages/huggingface_hub/hub_mixin.py", line 771, in _from_pretrained model = cls(**model_kwargs) File "/home/user/app/mast3r.py", line 22, in __init__ super().__init__(**kwargs) File "/usr/local/lib/python3.10/site-packages/mini_dust3r/model.py", line 83, in __init__ self.set_downstream_head( File "/home/user/app/mast3r.py", line 43, in set_downstream_head from catmlp_dpt_head import Cat_MLP_LocalFeatures_DPT_Pts3d, postprocess # Corrected import ModuleNotFoundError: No module named 'catmlp_dpt_head'

Container logs:

Fetching error logs...