rt-j-onnx / config.json
dhenneberger1's picture
Add ONNX model metadata
ea096c4 verified
Raw
History Blame Contribute Delete
297 Bytes
{
"architectures": ["RTJModel"],
"library_name": "relational-transformers",
"source_model": "RelativeDB/rt-j-fp16",
"task_type": "classification",
"model_file": "model.onnx",
"d_text": 384,
"d_model": 512,
"num_blocks": 12,
"num_heads": 8,
"dynamic_axes": ["batch", "cells"]
}