hidden_cfg = {
        "num_attention_heads":
            2,
        "intermediate_size":
            3072,