“WadoodAbdul”
commited on
Commit
·
9f897d4
1
Parent(s):
54f77a5
updated prompts
Browse files- src/display/utils.py +3 -2
- src/submission/submit.py +1 -1
src/display/utils.py
CHANGED
@@ -158,8 +158,9 @@ class Precision(Enum):
|
|
158 |
class PromptTemplateName(Enum):
|
159 |
UniversalNERTemplate = "universal_ner"
|
160 |
LLMHTMLHighlightedSpansTemplate = "llm_html_highlighted_spans"
|
161 |
-
|
162 |
-
|
|
|
163 |
|
164 |
|
165 |
|
|
|
158 |
class PromptTemplateName(Enum):
|
159 |
UniversalNERTemplate = "universal_ner"
|
160 |
LLMHTMLHighlightedSpansTemplate = "llm_html_highlighted_spans"
|
161 |
+
LLMHTMLHighlightedSpansTemplateV1 = "llm_html_highlighted_spans_v1"
|
162 |
+
LLamaNERTemplate = "llama_70B_ner"
|
163 |
+
# MixtralNERTemplate = "mixtral_ner_v0.3"
|
164 |
|
165 |
|
166 |
|
src/submission/submit.py
CHANGED
@@ -137,7 +137,7 @@ def add_new_eval(
|
|
137 |
return styled_error("Prompt template name is invalid")
|
138 |
inference_config = {
|
139 |
**inference_config,
|
140 |
-
"
|
141 |
}
|
142 |
case "GLiNER Encoder":
|
143 |
try:
|
|
|
137 |
return styled_error("Prompt template name is invalid")
|
138 |
inference_config = {
|
139 |
**inference_config,
|
140 |
+
"prompt_template_identifier": prompt_template_name,
|
141 |
}
|
142 |
case "GLiNER Encoder":
|
143 |
try:
|