code: def get_torch_model_size(
      model: Union[
          torch.nn.Module, torch.jit.ScriptModule, torch.fx.GraphModule
      ]
  ):
      param_size = 0
      for param in model.parameters():
          param_size += param.nelement() * param.element_size()
      buffer_size = 0
      for buffer in model.buffers():
          buffer_size += buffer.nelement() * buffer.element_size()
  
      return param_size + buffer_size
  
file_path: nebuly/optimization/nebullvm/nebullvm/tools/pytorch.py
imports: from pathlib import Path
  from typing import List, Tuple, Optional, Dict, Union, Sequence
  from loguru import logger
  from nebullvm.core.models import Device, DataType, DeviceType, InputInfo
  from nebullvm.optional_modules.torch import torch, DataLoader
  from nebullvm.tools.data import DataManager
  from nebullvm.tools.diffusers import get_default_dynamic_info
label_desc: Use image node_id 10 for calling a global function with example usage: get_torch_model_size(model) and returns: unknown