code: def data_gen():
      # Generated from following code snippet
      #
      # from transformers import GPT2Tokenizer
      # input = 'Hello, my dog is cute is cute' (last two words repeated to satisfy length requirement)
      # tokenized_input = tokenizer(input, return_tensors='pt')
      # input_ids = tokenized_input['input_ids']
      # attention_mask = tokenized_input['attention_mask']
      input_ids = torch.tensor(
          [[15496, 11, 616, 3290, 318, 13779, 318, 13779]],
          dtype=torch.int64,
      )
      attention_mask = torch.tensor(
          [[1, 1, 1, 1, 1, 1, 1, 1]], dtype=torch.int64
      )
      return dict(input_ids=input_ids, attention_mask=attention_mask)
  
file_path: colossalai/tests/kit/model_zoo/transformers/gpt.py
imports: import copy
  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