code: def _init_inference_stages_from_directory(
      directory: str,
      device: TorchDevice,
  ) -> Dict[str, torch.nn.Module]:
      """Initializes inference stage modules from directory."""
      stage_to_filenames = {
          stage: get_filename_from_stage(stage, device)
          for stage in [PREPROCESSOR, PREDICTOR, POSTPROCESSOR]
      }
  
      stage_to_module = {}
      for stage in [PREPROCESSOR, PREDICTOR, POSTPROCESSOR]:
          stage_to_module[stage] = torch.jit.load(
              os.path.join(directory, stage_to_filenames[stage])
          )
          print(
              f"Loaded torchscript module for {stage} from {stage_to_filenames[stage]}."
          )
      return stage_to_module
  
imports: import logging
  import os
  from typing import Any, Dict, Optional, TYPE_CHECKING, Union
  import pandas
  import torch
  from torch import nn
  from ludwig.constants import NAME, POSTPROCESSOR, PREDICTOR, PREPROCESSOR, TYPE
  from ludwig.data.postprocessing import convert_dict_to_df
  from ludwig.data.preprocessing import load_metadata
  from ludwig.features.feature_registries import get_input_type_registry
  from ludwig.features.feature_utils import get_module_dict_key_from_name, get_name_from_module_dict_key
  from ludwig.globals import MODEL_HYPERPARAMETERS_FILE_NAME, TRAIN_SET_METADATA_FILE_NAME
  from ludwig.types import ModelConfigDict, TrainingSetMetadataDict
  from ludwig.utils import output_feature_utils
  from ludwig.utils.data_utils import load_json, save_json
  from ludwig.utils.inference_utils import get_filename_from_stage, to_inference_module_input_from_dataframe
  from ludwig.utils.misc_utils import get_from_registry
  from ludwig.utils.output_feature_utils import get_feature_name_from_concat_name, get_tensor_name_from_concat_name
  from ludwig.utils.torch_utils import DEVICE
  from ludwig.utils.types import TorchDevice, TorchscriptPreprocessingInput
label_desc: Use image node_id 2 for calling a global function with example usage: _init_inference_stages_from_directory(directory, device) and returns: stage_to_module