code: def model_name(request):
      return request.param
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/inference/test_checkpoint_sharding.py
imports: import os
  import pytest
  import torch
  import deepspeed
  from deepspeed.model_implementations import DeepSpeedTransformerInference
  from transformers import AutoConfig, AutoModelForCausalLM
  from unit.common import DistributedTest, DistributedFixture
label_desc: Use image node_id 2 for calling a global function with example usage: model_name(request) and returns: request