“WadoodAbdul” commited on
Commit
9f897d4
·
1 Parent(s): 54f77a5

updated prompts

Browse files
Files changed (2) hide show
  1. src/display/utils.py +3 -2
  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
- LLamaNERTemplate = "llama_70B_ner_v0.3"
162
- MixtralNERTemplate = "mixtral_ner_v0.3"
 
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
- "prompt_template_name": prompt_template_name,
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: