configs¶

Classes¶

`ModelConfig`
`PromptConfig`

Module Contents¶

class configs.ModelConfig(tokenizer, use_chat_template, chat_template_type, gen_params, model_name, device)¶

tokenizer¶

use_chat_template¶

chat_template_type¶

gen_params¶

model_name¶

device¶

class configs.PromptConfig(system_prompt, user_prompt)¶

system_prompt¶

user_prompt¶