rwkv integration completed

This commit is contained in:
Concedo 2023-05-28 00:48:56 +08:00
parent 55e0fbf024
commit 5d9f5b28a6
4 changed files with 99 additions and 35 deletions

View file

@ -212,6 +212,10 @@ void print_tok_vec(std::vector<float> &embd)
{
fileformat = FileFormat::RWKV_1;
}
else if(temp==101)
{
fileformat = FileFormat::RWKV_2;
}
}
else if(magic == 0x67676a74) //v3 format ggjt
{