Boltz1.ipynb: "RuntimeError: operator torchvision::nms does not exist"

Using the Boltz1 script on Google Colab, I keep getting this error no matter what sample I run (default test or my own sequences) or which GPU type. It was working fine for me a few weeks ago.

Any ideas what's going on? Thanks in advance!

Traceback (most recent call last):
File "/usr/local/bin/boltz", line 5, in
from boltz.main import cli
File "/usr/local/lib/python3.11/dist-packages/boltz/main.py", line 16, in
from pytorch_lightning import Trainer, seed_everything
File "/usr/local/lib/python3.11/dist-packages/pytorch_lightning/init.py", line 27, in
from pytorch_lightning.callbacks import Callback # noqa: E402
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/pytorch_lightning/callbacks/init.py", line 14, in
from pytorch_lightning.callbacks.batch_size_finder import BatchSizeFinder
File "/usr/local/lib/python3.11/dist-packages/pytorch_lightning/callbacks/batch_size_finder.py", line 26, in
from pytorch_lightning.callbacks.callback import Callback
File "/usr/local/lib/python3.11/dist-packages/pytorch_lightning/callbacks/callback.py", line 22, in
from pytorch_lightning.utilities.types import STEP_OUTPUT
File "/usr/local/lib/python3.11/dist-packages/pytorch_lightning/utilities/types.py", line 36, in
from torchmetrics import Metric
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/init.py", line 37, in
from torchmetrics import functional # noqa: E402
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/init.py", line 56, in
from torchmetrics.functional.image._deprecated import (
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/image/init.py", line 14, in
from torchmetrics.functional.image.arniqa import arniqa
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/image/arniqa.py", line 31, in
from torchvision import transforms
File "/usr/local/lib/python3.11/dist-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torchvision/_meta_registrations.py", line 163, in
@torch.library.register_fake("torchvision::nms")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torch/library.py", line 1023, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File "/usr/local/lib/python3.11/dist-packages/torch/library.py", line 214, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torch/_library/fake_impl.py", line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: operator torchvision::nms does not exist

Start with Boltz1.ipynb and the traceback's import chain through pytorch_lightning, torchmetrics, and torchvision. Inspect the installed torch and torchvision versions in Google Colab and reproduce the import failure; done means the notebook starts without the torchvision::nms error.