mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-07-01 02:45:33 +02:00
special tokens in prompt template
This commit is contained in:
@@ -21,9 +21,10 @@ class LanguageModel(AbstractLanguageModel):
|
||||
llm = TextGenerationFoundationModel().build()
|
||||
|
||||
# Phi-3 specific prompt template
|
||||
template = """<|user|>
|
||||
template = """<|system|>
|
||||
Act as a conversational agent to respond to the end user's prompt.
|
||||
Question: {question}
|
||||
<|user|>
|
||||
Question: {question}<|end|>
|
||||
<|assistant|>
|
||||
"""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user