Add support for 30B model and 65B, if it is made in the future (#104)
This commit is contained in:
parent
7636ddb9e4
commit
428aa7025a
1 changed files with 2 additions and 2 deletions
4
chat.cpp
4
chat.cpp
|
@ -32,8 +32,8 @@
|
|||
static const std::map<int, int> LLAMA_N_PARTS = {
|
||||
{ 4096, 1 },
|
||||
{ 5120, 1 },
|
||||
{ 6656, 4 },
|
||||
{ 8192, 8 },
|
||||
{ 6656, 1 },
|
||||
{ 8192, 1 },
|
||||
};
|
||||
|
||||
// default hparams (LLaMA 7B)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue