18 lines
521 B
Python
18 lines
521 B
Python
class TemplateQwen():
|
|
# --- Prompt Template Configuration (ChatML) ---
|
|
IM_START = "<|im_start|>"
|
|
IM_END = "<|im_end|>"
|
|
ROLE_SYSTEM = "system"
|
|
ROLE_USER = "user"
|
|
ROLE_ASSISTANT = "assistant"
|
|
|
|
class TemplateGemma():
|
|
# --- Prompt Template Configuration (Gemma3) ---
|
|
IM_START = "<start_of_turn>"
|
|
IM_END = "<end_of_turn>"
|
|
ROLE_SYSTEM = "user" # Gemma has no system role
|
|
ROLE_USER = "user"
|
|
ROLE_ASSISTANT = "model"
|
|
|
|
agent_template = TemplateQwen()
|
|
repl_template = TemplateGemma() |