AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-10-01 01:06:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Aligned llama implementation style to other implementations

Browse Source
This commit is contained in:
niansa 2023-05-20 00:49:00 +02:00 committed by Adam Treat
parent 5d723221b7
commit 6fad98e28f
2 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
gpt4all-backend/llamamodel.cpp
View File

@ -28,6 +28,7 @@
#include <llama.h>
#include <ggml.h>
namespace {
const char *modelType_ = "LLaMA";
}
@ -48,16 +49,7 @@ struct gpt_params {
bool use_mlock = false; // use mlock to keep model in memory
};
struct LLamaPrivate {
const std::string modelPath;
bool modelLoaded;
llama_context *ctx = nullptr;
llama_context_params params;
int64_t n_threads = 0;
bool empty = true;
};
static int llama_sample_top_p_top_k(
int llama_sample_top_p_top_k(
llama_context *ctx,
const llama_token *last_n_tokens_data,
int last_n_tokens_size,
@ -85,6 +77,15 @@ static int llama_sample_top_p_top_k(
return llama_sample_token(ctx, &candidates_p);
}
struct LLamaPrivate {
const std::string modelPath;
bool modelLoaded;
llama_context *ctx = nullptr;
llama_context_params params;
int64_t n_threads = 0;
bool empty = true;
};
LLamaModel::LLamaModel()
: d_ptr(new LLamaPrivate) {
modelType = modelType_;

1
gpt4all-backend/llamamodel_old.cpp
View File

@ -28,6 +28,7 @@
#include <llama.h>
#include <ggml.h>
namespace {
const char *modelType_ = "LLaMA";
}