ember.cpp 0.1.0
C++ inference engine for Llama-family and paperkiln models (CPU, CUDA, WebAssembly)
Loading...
Searching...
No Matches
tinyllama::TinyLlamaSession Member List

This is the complete list of members for tinyllama::TinyLlamaSession, including all inherited members.

batch_generation_parallel(const std::vector< int > &current_tokens, const std::vector< int > &token_positions, const std::vector< int > &original_sequence_indices, std::vector< std::vector< float > > &batch_logits)tinyllama::TinyLlamaSessionprivate
batch_prefill_parallel(const std::vector< std::vector< int > > &all_tokens, const std::vector< int > &prompt_lengths, std::vector< std::vector< float > > &batch_final_logits)tinyllama::TinyLlamaSessionprivate
bpe_tinyllama::TinyLlamaSessionprivate
bpe_decode(const std::vector< int > &ids) consttinyllama::TinyLlamaSessioninline
bpe_encode(const std::string &text) consttinyllama::TinyLlamaSessioninline
bpe_prompt_ids(const std::string &text) consttinyllama::TinyLlamaSessioninline
bpe_token_count(const std::string &text) consttinyllama::TinyLlamaSessioninline
chat_format() consttinyllama::TinyLlamaSessioninline
chat_template() consttinyllama::TinyLlamaSessioninline
chat_template_why() consttinyllama::TinyLlamaSessioninline
chat_template_why_tinyllama::TinyLlamaSessionprivate
config_tinyllama::TinyLlamaSessionprivate
eos_token_id_tinyllama::TinyLlamaSessionprivate
generate(const std::string &prompt, int steps=128, float temperature=0.1f, int top_k=40, float top_p=0.9f, const std::string &system_prompt="", bool apply_q_a_format=false)tinyllama::TinyLlamaSession
generate_batch(const std::vector< std::string > &prompts, int steps=128, float temperature=0.1f, int top_k=40, float top_p=0.9f, const std::string &system_prompt="", bool apply_q_a_format=false)tinyllama::TinyLlamaSession
generated_stream_tinyllama::TinyLlamaSessionprivate
generated_text_for_api_return_tinyllama::TinyLlamaSessionprivate
get_config() consttinyllama::TinyLlamaSessioninline
get_kv_cache()tinyllama::TinyLlamaSessioninline
get_repetition_penalty() consttinyllama::TinyLlamaSessioninline
get_tokenizer() consttinyllama::TinyLlamaSessioninline
gguf_tokens() consttinyllama::TinyLlamaSession
ir_tinyllama::TinyLlamaSessionprivate
ir_doc_tinyllama::TinyLlamaSessionprivate
ir_document() consttinyllama::TinyLlamaSessioninline
is_bpe() consttinyllama::TinyLlamaSessioninline
is_chat_eos(int id) consttinyllama::TinyLlamaSessioninline
is_chat_stop(int id) consttinyllama::TinyLlamaSessioninline
kv_cache_tinyllama::TinyLlamaSessionprivate
kv_token_history_tinyllama::TinyLlamaSessionprivate
last_generated_ids() consttinyllama::TinyLlamaSessioninline
last_generated_ids_tinyllama::TinyLlamaSessionprivate
last_generated_tokens() consttinyllama::TinyLlamaSessioninline
last_generated_tokens_tinyllama::TinyLlamaSessionprivate
last_prompt_ids() consttinyllama::TinyLlamaSessioninline
last_prompt_ids_tinyllama::TinyLlamaSessionprivate
last_stop_id() consttinyllama::TinyLlamaSessioninline
last_stop_id_tinyllama::TinyLlamaSessionprivate
last_stop_reason() consttinyllama::TinyLlamaSessioninline
last_stop_reason_tinyllama::TinyLlamaSessionprivate
last_ttft() consttinyllama::TinyLlamaSessioninline
last_ttft_tinyllama::TinyLlamaSessionprivate
max_batch_size_tinyllama::TinyLlamaSessionprivate
model_tinyllama::TinyLlamaSessionprivate
operator=(const TinyLlamaSession &)=deletetinyllama::TinyLlamaSessionprivate
repeat_last_n_tinyllama::TinyLlamaSessionprivate
repeat_penalty_tinyllama::TinyLlamaSessionprivate
repetition_penalty_tinyllama::TinyLlamaSessionprivate
rng_tinyllama::TinyLlamaSessionprivate
sample_step(std::vector< float > &logits, const std::vector< int > &context, float temperature, int top_k, float top_p)tinyllama::TinyLlamaSession
score_context() consttinyllama::TinyLlamaSession
score_ids_tinyllama::TinyLlamaSessionprivate
score_reuse_ok() consttinyllama::TinyLlamaSession
score_rows(const std::vector< int > &ids, int first_row, bool reuse_prefix, int *reused=nullptr)tinyllama::TinyLlamaSession
score_vocab() consttinyllama::TinyLlamaSession
scripted_replies_tinyllama::TinyLlamaSessionprivate
set_repeat_penalty(float p, int last_n)tinyllama::TinyLlamaSessioninline
set_repetition_penalty(float p)tinyllama::TinyLlamaSessioninline
set_scripted_replies(std::vector< std::vector< int > > r)tinyllama::TinyLlamaSessioninline
set_seed(unsigned seed)tinyllama::TinyLlamaSessioninline
set_token_callback(std::function< void(int)> cb)tinyllama::TinyLlamaSessioninline
state_k() consttinyllama::TinyLlamaSessioninline
state_k_tinyllama::TinyLlamaSessionprivate
state_vocab_tinyllama::TinyLlamaSessionprivate
threads_tinyllama::TinyLlamaSessionprivate
TinyLlamaSession(const std::string &model_path, const std::string &tokenizer_path, int threads=1, int num_gpu_layers_from_cli=0, bool cli_use_mmap=true, bool use_kv_quant=false, bool use_batch_generation=false, int max_batch_size=1)tinyllama::TinyLlamaSession
TinyLlamaSession(const TinyLlamaSession &)=deletetinyllama::TinyLlamaSessionprivate
token_callback_tinyllama::TinyLlamaSessionprivate
tokenizer_tinyllama::TinyLlamaSessionprivate
use_batch_generation_tinyllama::TinyLlamaSessionprivate
word_template_tinyllama::TinyLlamaSessionprivate
~TinyLlamaSession()tinyllama::TinyLlamaSession