root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-09-23 05:26:19 +00:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Merge f9c2155158 into c35e586ea5

Browse Source
This commit is contained in:
Daniel Bevenius 2024-09-22 10:57:19 -06:00 committed by GitHub
commit b1c09b186e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/baby-llama/baby-llama.cpp
View File

@ -11,6 +11,8 @@
#pragma warning(disable: 4244 4267) // possible loss of data #pragma warning(disable: 4244 4267) // possible loss of data
#endif #endif
namespace {
#ifdef LLAMA_DEFAULT_RMS_EPS #ifdef LLAMA_DEFAULT_RMS_EPS
constexpr float rms_norm_eps = LLAMA_DEFAULT_RMS_EPS; constexpr float rms_norm_eps = LLAMA_DEFAULT_RMS_EPS;
#else #else
@ -1432,7 +1434,7 @@ static struct ggml_tensor * cross_entropy_loss(
ggml_new_f32(ctx, eps))))))); ggml_new_f32(ctx, eps)))))));
} }
int main(int argc, char ** argv) { int baby_llama_main(int argc, char ** argv) {
if (argc < 1) { if (argc < 1) {
fprintf(stderr, "usage: %s\n", argv[0]); fprintf(stderr, "usage: %s\n", argv[0]);
@ -1637,3 +1639,9 @@ int main(int argc, char ** argv) {
return 0; return 0;
} }
}
int main(int argc, char ** argv) {
return baby_llama_main(argc, argv);
}