code: def gpt2_14b(checkpoint=True):
      return GPTLMModel(
          hidden_size=4096,
          num_layers=70,
          num_attention_heads=16,
          checkpoint=checkpoint,
      )
  
file_path: ColossalAI/examples/language/gpt/gemini/commons/model_zoo.py
imports: from torch import nn
  from transformers import GPT2Config, GPT2LMHeadModel
label_desc: Use image node_id 4 for calling a global function with example usage: gpt2_14b(checkpoint) and returns: GPTLMModel