code: def bench_one_case(case):
      if case.model == "facebook/opt-6.7b" and case.device == "gpu":
          cut_gen_len = None
      else:
          cut_gen_len = 16
      dummy = True
  
      if case.device == "gpu":
          cpu = disk = False
      elif case.device == "cpu":
          cpu, disk = True, False
      elif case.device == "disk":
          cpu, disk = False, True
  
      use_deepspeed = case.library == "ds"
  
      run_huggingface(
          case.model,
          case.prompt_len,
          case.gen_len,
          cut_gen_len,
          case.batch_size,
          case.num_nodes,
          case.num_gpus_per_node,
          use_ds=use_deepspeed,
          cpu=cpu,
          disk=disk,
          dummy=dummy,
      )
  
imports: import argparse
  from dataclasses import dataclass
  import time
  from flexgen.utils import run_cmd
label_desc: Use image node_id 2 for calling a global function with example usage: bench_one_case(case) without return types