-record(bert_model, {
    embedding :: viva_tensor@models@bert:bert_embedding(),
    blocks :: list(viva_tensor@models@bert:bert_block())
}).