SAI_35M / source_config /35M.json
thongbuind's picture
Use original SAI source for inference
fb6b7f8 verified
Raw
History Blame Contribute Delete
173 Bytes
{
"d_model": 512,
"num_heads": 8,
"num_kv_heads": 8,
"num_layers": 8,
"ff_dim": 2048,
"batch_size": 70,
"freeze": [0, 1],
"num_workers": 1
}