code: def remove_random_ltd_state_dict(state_dict):
      new_state_dict = OrderedDict()
      for key, value in state_dict.items():
          if ".random_ltd_layer" in key:
              new_key = "".join(key.split(".random_ltd_layer"))
          else:
              new_key = key
          new_state_dict[new_key] = value
      return new_state_dict
  
imports: from .basic_layer import RandomLayerTokenDrop
  from collections import OrderedDict
  from deepspeed.compression.helper import recursive_getattr, recursive_setattr
label_desc: Use image node_id 3 for calling a global function with example usage: remove_random_ltd_state_dict(state_dict) and returns: new_state_dict