code: def get_size_policy(min_params=1e8):
      num_wrap_policy = functools.partial(
          size_based_auto_wrap_policy, min_num_params=min_params
      )
      return num_wrap_policy
  
imports: import functools
  from transformers.models.llama.modeling_llama import LlamaDecoderLayer
  from torch.distributed.fsdp.wrap import transformer_auto_wrap_policy, size_based_auto_wrap_policy
label_desc: Use image node_id 1 for calling a global function with example usage: get_size_policy(min_params) and returns: num_wrap_policy