vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

read static_input_file

Browse source
This commit is contained in:
JohannesGaessler 2024-02-07 19:19:39 +01:00
parent 51b7317a65
commit 6d47013d81
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
examples/lookup-static/lookup-static.cpp
View file

@ -3,11 +3,24 @@
#include <cmath> #include <cmath>
#include <cstdio> #include <cstdio>
#include <fstream>
#include <iostream>
#include <stdlib.h> #include <stdlib.h>
#include <string> #include <string>
#include <vector> #include <vector>
int main(int argc, char ** argv){ int main(int argc, char ** argv){
const char * static_input_file = "./wikitext-2-raw/wiki.test.raw";
std::ifstream file(static_input_file);
if (!file) {
fprintf(stderr, "error: failed to open file '%s'\n", static_input_file);
exit(1);
}
std::string static_input;
std::copy(std::istreambuf_iterator<char>(file), std::istreambuf_iterator<char>(), back_inserter(static_input));
if (!static_input.empty() && static_input.back() == '\n') {
static_input.pop_back();
}
gpt_params params; gpt_params params;
if (!gpt_params_parse(argc, argv, params)) { if (!gpt_params_parse(argc, argv, params)) {