configs
=======

.. py:module:: configs


Classes
-------

.. autoapisummary::

   configs.ModelConfig
   configs.PromptConfig


Module Contents
---------------

.. py:class:: ModelConfig(tokenizer, use_chat_template, chat_template_type, gen_params, model_name, device)

   .. py:attribute:: tokenizer


   .. py:attribute:: use_chat_template


   .. py:attribute:: chat_template_type


   .. py:attribute:: gen_params


   .. py:attribute:: model_name


   .. py:attribute:: device


.. py:class:: PromptConfig(system_prompt, user_prompt)

   .. py:attribute:: system_prompt


   .. py:attribute:: user_prompt