lmflow.utils.llava_conversation_lib#

Attributes#

Classes#

SeparatorStyle

Different separator style.

Conversation

A class that keeps all conversation history.

Module Contents#

class lmflow.utils.llava_conversation_lib.SeparatorStyle(*args, **kwds)[source]#

Bases: enum.Enum

Different separator style.

SINGLE[source]#
TWO[source]#
MPT[source]#
PLAIN[source]#
LLAMA_2[source]#
class lmflow.utils.llava_conversation_lib.Conversation[source]#

A class that keeps all conversation history.

system: str[source]#
roles: List[str][source]#
messages: List[List[str]][source]#
offset: int[source]#
sep_style: SeparatorStyle[source]#
sep: str = '###'[source]#
sep2: str = None[source]#
version: str = 'Unknown'[source]#
skip_next: bool = False[source]#
get_prompt()[source]#
append_message(role, message)[source]#
get_images(return_pil=False)[source]#
to_gradio_chatbot()[source]#
copy()[source]#
dict()[source]#
lmflow.utils.llava_conversation_lib.conv_vicuna_v0[source]#
lmflow.utils.llava_conversation_lib.conv_vicuna_v1[source]#
lmflow.utils.llava_conversation_lib.conv_llama_2[source]#
lmflow.utils.llava_conversation_lib.conv_llava_llama_2[source]#
lmflow.utils.llava_conversation_lib.conv_mpt[source]#
lmflow.utils.llava_conversation_lib.conv_llava_plain[source]#
lmflow.utils.llava_conversation_lib.conv_llava_v0[source]#
lmflow.utils.llava_conversation_lib.conv_llava_v0_mmtag[source]#
lmflow.utils.llava_conversation_lib.conv_llava_v1[source]#
lmflow.utils.llava_conversation_lib.conv_llava_v1_mmtag[source]#
lmflow.utils.llava_conversation_lib.default_conversation[source]#
lmflow.utils.llava_conversation_lib.conv_templates[source]#