| #ifndef LITERARY_INFER_H |
| #define LITERARY_INFER_H |
|
|
| #include <stdint.h> |
|
|
| enum { |
| LM_HOLO_DISABLED = 0, |
| LM_HOLO_FLAT = 1, |
| LM_HOLO_PARTITIONED = 2 |
| }; |
|
|
| int lm_load(const unsigned char *data, int length); |
| void lm_reset(void); |
| void lm_seed(uint32_t seed); |
| int lm_feed(int token); |
| int lm_sample(float temperature, int top_k, float repetition_penalty); |
| float lm_probability(int token); |
| int lm_get_context(void); |
| int lm_get_position(void); |
| int lm_get_update(void); |
| int lm_get_parameters(void); |
|
|
| void lm_holo_reset(void); |
| int lm_holo_set_mode(int mode); |
| int lm_holo_get_mode(void); |
| int lm_holo_remember(const unsigned char *text, int length); |
| int lm_holo_recall(const unsigned char *text, int length); |
| float lm_holo_get_score(void); |
| int lm_holo_get_count(void); |
|
|
| #endif |
|
|