File size: 393 Bytes
5303207 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
#ifndef MAIN_H
#define MAIN_H
#ifdef __cplusplus
extern "C" {
#endif
typedef struct Main Main;
Main *build_main(char* checkpoint_path, char* tokenizer_path, float temperature, float topp, int steps,
char* prompt, unsigned long long rng_seed, char* mode, char* system_prompt);
void free_main(Main *m);
char *run_main(Main *m);
#ifdef __cplusplus
}
#endif
#endif // MAIN_H
|