'No module named 'torch._custom_ops' upon training model

[TheFlano23]

Upon inputting the dataset directory and training the Dreambooth model I receive the following traceback:

Traceback (most recent call last): File "/usr/local/bin/accelerate", line 5, in <module> from accelerate.commands.accelerate_cli import main File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/accelerate_cli.py", line 19, in <module> from accelerate.commands.estimate import estimate_command_parser File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/estimate.py", line 34, in <module> import timm File "/usr/local/lib/python3.10/dist-packages/timm/__init__.py", line 2, in <module> from .models import create_model, list_models, is_model, list_modules, model_entrypoint, \ File "/usr/local/lib/python3.10/dist-packages/timm/models/__init__.py", line 1, in <module> from .beit import * File "/usr/local/lib/python3.10/dist-packages/timm/models/beit.py", line 49, in <module> from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD File "/usr/local/lib/python3.10/dist-packages/timm/data/__init__.py", line 5, in <module> from .dataset import ImageDataset, IterableImageDataset, AugMixDataset File "/usr/local/lib/python3.10/dist-packages/timm/data/dataset.py", line 12, in <module> from .parsers import create_parser File "/usr/local/lib/python3.10/dist-packages/timm/data/parsers/__init__.py", line 1, in <module> from .parser_factory import create_parser File "/usr/local/lib/python3.10/dist-packages/timm/data/parsers/parser_factory.py", line 3, in <module> from .parser_image_folder import ParserImageFolder File "/usr/local/lib/python3.10/dist-packages/timm/data/parsers/parser_image_folder.py", line 11, in <module> from timm.utils.misc import natural_key File "/usr/local/lib/python3.10/dist-packages/timm/utils/__init__.py", line 2, in <module> from .checkpoint_saver import CheckpointSaver File "/usr/local/lib/python3.10/dist-packages/timm/utils/checkpoint_saver.py", line 15, in <module> from .model import unwrap_model, get_state_dict File "/usr/local/lib/python3.10/dist-packages/timm/utils/model.py", line 8, in <module> from torchvision.ops.misc import FrozenBatchNorm2d File "/usr/local/lib/python3.10/dist-packages/torchvision/__init__.py", line 10, in <module> from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip File "/usr/local/lib/python3.10/dist-packages/torchvision/_meta_registrations.py", line 4, in <module> import torch._custom_ops ModuleNotFoundError: No module named 'torch._custom_ops'

Any help would be appreciated