code: def data_gen():
      input_ids = torch.Tensor(
          [[1, 15043, 29892, 590, 11203, 338, 274, 1082]]
      ).long()
      attention_mask = torch.Tensor([[1, 1, 1, 1, 1, 1, 1, 1]]).long()
      return dict(input_ids=input_ids, attention_mask=attention_mask)
  
imports: import torch
  import transformers
  from ..registry import ModelAttribute, model_zoo
label_desc: Use image node_id 1 for calling a global function with example usage: data_gen() and returns: dict