hotfix for rwkv

This commit is contained in:
Concedo 2023-06-13 23:38:41 +08:00
parent 9db2ec068f
commit 82cf97ce92
3 changed files with 3 additions and 3 deletions

View file

@ -479,7 +479,7 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
}
else
{
n_batch = 10; //use sequence mode to speedup
n_batch = 8; //use sequence mode to speedup
//setup buffers for rwkv state
auto padding = 512u;