code: def query(model, task):
      if task == "text-generation":
          return "DeepSpeed is"
      elif task == "fill-mask":
          if "roberta" in model:
              return "I am a <mask> model"
          else:
              return "I am a [MASK] model"
      else:
          raise NotImplementedError
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/inference/test_model_profiling.py
imports: import os
  import time
  import pytest
  import torch
  import deepspeed
  from transformers import pipeline
  from unit.common import DistributedTest
label_desc: Use image node_id 1 for calling a global function with example usage: query(model, task) and returns: str, str, str