Add support for 30B model and 65B, if it is made in the future (#104)

This commit is contained in:
Pi 2023-03-21 13:55:24 -07:00 committed by GitHub
parent 7636ddb9e4
commit 428aa7025a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)