Final touches

This commit is contained in:
Georgi Gerganov 2023-03-10 21:50:46 +02:00
parent 775328064e
commit 319cdb3e1f
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
5 changed files with 32 additions and 32 deletions

View file

@ -15,12 +15,12 @@
struct gpt_params {
int32_t seed = -1; // RNG seed
int32_t n_threads = std::min(4, (int32_t) std::thread::hardware_concurrency());
int32_t n_predict = 200; // new tokens to predict
int32_t n_predict = 128; // new tokens to predict
// sampling parameters
int32_t top_k = 100;
int32_t top_k = 40;
float top_p = 0.95f;
float temp = 0.8f;
float temp = 0.80f;
int32_t n_batch = 8; // batch size for prompt processing