happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update run.cc, CMakeLists and README for incompatible code, dependency changes and argument updates

This commit is contained in:
KaranocaVe 2025-07-31 00:59:16 +08:00
parent 4a924f1794
commit 0ea118ebbe
3 changed files with 74 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/hello_world/CMakeLists.txt
View File

@ -14,12 +14,13 @@
cmake_minimum_required(VERSION 3.11)
project(hello_world)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
include(FetchContent)
FetchContent_Declare(highway GIT_REPOSITORY https://github.com/google/highway.git GIT_TAG c5bebf84ad01edec97e336f5c97ca4e0df6b4d06)
FetchContent_Declare(highway GIT_REPOSITORY https://github.com/google/highway.git GIT_TAG bfc36a6e633af94e63ac4b91c687bf0354cb24e0)
FetchContent_MakeAvailable(highway)
FetchContent_Declare(sentencepiece GIT_REPOSITORY https://github.com/google/sentencepiece GIT_TAG 53de76561cfc149d3c01037f0595669ad32a5e7c)
FetchContent_Declare(sentencepiece GIT_REPOSITORY https://github.com/google/sentencepiece GIT_TAG 9045b2f60fa2b323dfac0eaef8fc17565036f9f9)
FetchContent_MakeAvailable(sentencepiece)
@ -33,7 +34,7 @@ if (BUILD_MODE STREQUAL "local")
# Relative path to gemma.cpp from examples/hello_world/build/
FetchContent_Declare(gemma SOURCE_DIR ../../..)
else()
FetchContent_Declare(gemma GIT_REPOSITORY https://github.com/google/gemma.cpp.git GIT_TAG a9aa63fd2ea6b786ed0706d619588bfe2d43370e)
FetchContent_Declare(gemma GIT_REPOSITORY https://github.com/google/gemma.cpp.git GIT_TAG 4a924f179448dc83e46a2af9520c61b4ef56174c)
endif()
FetchContent_MakeAvailable(gemma)

2
examples/hello_world/README.md
View File

@ -41,7 +41,7 @@ gemma.cpp specifying the tokenizer, compressed weights file, and model type, for
example:
```sh
./hello_world --tokenizer tokenizer.spm --compressed_weights 2b-it-sfp.sbs --model 2b-it
./hello_world --tokenizer tokenizer.spm --weights 2b-it-sfp.sbs --model 2b-it
```
Should print a greeting to the terminal:

10
examples/hello_world/run.cc
View File

@ -32,8 +32,7 @@
#include "hwy/base.h"
#include "hwy/contrib/thread_pool/thread_pool.h"
int main(int argc, char** argv) {
{
int main(int argc, char **argv) { {
// Placeholder for internal init, do not modify.
}
@ -43,7 +42,7 @@ int main(int argc, char** argv) {
if (gcpp::HasHelp(argc, argv)) {
loader.Help();
return 0;
} else if (const char* error = loader.Validate()) {
} else if (const char *error = loader.Validate()) {
loader.Help();
HWY_ABORT("\nInvalid args: %s", error);
}
@ -99,9 +98,10 @@ int main(int argc, char** argv) {
.verbosity = 0,
.stream_token = stream_token,
.accept_token =
std::function<bool(int, float)>(
[&](int token, float /* prob */) {
return !reject_tokens.contains(token);
},
return reject_tokens.find(token) == reject_tokens.end();
}),
};
model.Generate(runtime_config, tokens, 0, kv_cache, timing_info);
}