mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2024-09-22 21:16:20 +00:00
Merge f9c2155158
into c35e586ea5
This commit is contained in:
commit
b1c09b186e
@ -11,6 +11,8 @@
|
||||
#pragma warning(disable: 4244 4267) // possible loss of data
|
||||
#endif
|
||||
|
||||
namespace {
|
||||
|
||||
#ifdef LLAMA_DEFAULT_RMS_EPS
|
||||
constexpr float rms_norm_eps = LLAMA_DEFAULT_RMS_EPS;
|
||||
#else
|
||||
@ -1432,7 +1434,7 @@ static struct ggml_tensor * cross_entropy_loss(
|
||||
ggml_new_f32(ctx, eps)))))));
|
||||
}
|
||||
|
||||
int main(int argc, char ** argv) {
|
||||
int baby_llama_main(int argc, char ** argv) {
|
||||
if (argc < 1) {
|
||||
fprintf(stderr, "usage: %s\n", argv[0]);
|
||||
|
||||
@ -1637,3 +1639,9 @@ int main(int argc, char ** argv) {
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
int main(int argc, char ** argv) {
|
||||
return baby_llama_main(argc, argv);
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user