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

fixed add end condition for generating

Browse source
This commit is contained in:
ningshanwutuobang 2023-06-07 23:44:54 +08:00
parent ba1f617d7d
commit 6ed4893391
2 changed files with 22 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

8
examples/embd_input/embd_input.py
View file

@ -33,6 +33,7 @@ class MyModel:
s = libc.sampling(self.model)
return s
if __name__ == "__main__":
model = MyModel(["main", "--model", "../llama.cpp/models/ggml-vic13b-q4_1.bin", "-c", "2048"])
# print(model)
model.eval_string("""user: what is the color of the flag of UN?""")
@ -44,5 +45,8 @@ model.eval_string("""assistant:""")
# model.eval_float(x)
# print(libc)
for i in range(50):
print(model.sampling().decode(), end="", flush=True)
for i in range(500):
tmp = model.sampling().decode()
if tmp == "":
break
print(tmp, end="", flush=True)

8
examples/embd_input/embd_input_test.cpp
View file

@ -1,6 +1,7 @@
#include "embd_input.h"
#include <stdlib.h>
#include <random>
#include <string.h>
int main(int argc, char** argv) {
@ -20,9 +21,12 @@ int main(int argc, char** argv) {
eval_string(mymodel, "assistant:");
// printf("eval float end\n");
eval_string(mymodel, mymodel->params.prompt.c_str());
for (int i=0;i < 50; i++) {
const char* tmp;
for (int i=0;i < 500; i++) {
// int id = sampling_id(mymodel);
printf("%s", sampling(mymodel)); // llama_token_to_str(mymodel->ctx, id));
tmp = sampling(mymodel);
if (strlen(tmp) == 0) break;
printf("%s", tmp); // llama_token_to_str(mymodel->ctx, id));
fflush(stdout);
// eval_id(mymodel, id);
}