class_docstr: Args:
code: def __init__(
      self, trunk: nn.Module, heads: Union[nn.ModuleDict, nn.Module]
  ):
      super().__init__()
      self.trunk = trunk
      self.heads = heads
      self.types = ["image", "video", "rgbd"]
      self.multimodal_model = False
      if isinstance(heads, nn.ModuleDict):
          self.multimodal_model = True
          assert [
              n in heads for n in self.types
          ], "All heads must be provided"
  
imports: from towhee.models.video_swin_transformer.video_swin_transformer import VideoSwinTransformer
  from typing import Any, Optional, Union
  import torch
  from torch import nn
  from torch.hub import load_state_dict_from_url
label_desc: Use image node_id 1 to create a new OmnivoreModel object from inherited base classes: nn with example: obj = OmnivoreModel(trunk, heads)