SAI_35M / source_config /100M.json
thongbuind's picture
Use original SAI source for inference
fb6b7f8 verified
Raw
History Blame Contribute Delete
178 Bytes
{
"d_model": 768,
"num_heads": 12,
"num_kv_heads": 6,
"num_layers": 12,
"ff_dim": 3072,
"batch_size": 20,
"freeze": [0, 1, 2],
"num_workers": 4
}