Demo run with docker container fails.

[Rektino]

I have followed the instructions for building the docker image, and getting the weights in place.
Running the demo script commands fails. Here is the console log :

(my) root@linux772:/workspace/Fast-FoundationStereo# python scripts/run_demo.py --model_dir weights/23-36-37/model_best_bp2_serialize.pth --left_file demo_data/left.png --right_file demo_data/right.png --intrinsic_file demo_data/K.txt --out_dir output/ --remove_invisible 0 --denoise_cloud 1 --scale 1 --get_pc 1 --valid_iters 8 --max_disp 192 --zfar 100
args:
{'corr_levels': 2, 'corr_radius': 4, 'hidden_dims': [128], 'low_memory': 0, 'max_disp': 192, 'mixed_precision': True, 'n_downsample': 2, 'n_gru_layers': 1, 'slow_fast_gru': False, 'valid_iters': 8, 'vit_size': 'vitl', 'model_dir': 'weights/23-36-37/model_best_bp2_serialize.pth', 'left_file': 'demo_data/left.png', 'right_file': 'demo_data/right.png', 'intrinsic_file': 'demo_data/K.txt', 'out_dir': 'output/', 'remove_invisible': 0, 'denoise_cloud': 1, 'denoise_nb_points': 30, 'denoise_radius': 0.03, 'scale': 1.0, 'hiera': 0, 'get_pc': 1, 'zfar': 100.0}
Traceback (most recent call last):
File "/workspace/Fast-FoundationStereo/scripts/run_demo.py", line 56, in
model = torch.load(args.model_dir, map_location='cpu', weights_only=False)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/serialization.py", line 1530, in load
return _load(
^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/serialization.py", line 2122, in _load
result = unpickler.load()
^^^^^^^^^^^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/serialization.py", line 2111, in find_class
return super().find_class(mod_name, name)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/workspace/Fast-FoundationStereo/scripts/../core/foundation_stereo.py", line 1, in
import torch,pdb,logging,timm
File "/opt/conda/envs/my/lib/python3.12/site-packages/timm/init.py", line 2, in
from .layers import (
File "/opt/conda/envs/my/lib/python3.12/site-packages/timm/layers/init.py", line 1, in
from ._fx import (
File "/opt/conda/envs/my/lib/python3.12/site-packages/timm/layers/_fx.py", line 8, in
from torchvision.models.feature_extraction import create_feature_extractor as _create_feature_extractor
File "/opt/conda/envs/my/lib/python3.12/site-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torchvision/_meta_registrations.py", line 163, in
@torch.library.register_fake("torchvision::nms")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/library.py", line 1063, in register
use_lib._register_fake(
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/library.py", line 211, in _register_fake
handle = entry.fake_impl.register(
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/conda/envs/my/lib/python3.12/site-packages/torch/_library/fake_impl.py", line 50, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: operator torchvision::nms does not exist