CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-08-16 00:00:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

support prompt template loading

Browse Source
This commit is contained in:
Adam Wilson
2025-07-07 21:38:42 -06:00
parent 911b629217
commit af75e9aabf
22 changed files with 291 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/common/constants.py
+5 -1
View File
@@ -2,4 +2,8 @@ class Constants:
ASSISTANT_TOKEN = "<|assistant|>"
END_TOKEN = "<|end|>"
SYSTEM_TOKEN = "<|system|>"
USER_TOKEN = "<|user|>"
USER_TOKEN = "<|user|>"
# prompt template IDs
class PromptTemplateIds:
PHI_3_MINI_4K_INSTRUCT_BASIC = "phi-3-mini-4k-instruct-basic"