Upload templates.py with huggingface_hub

templates.py

@@ -109,6 +109,7 @@ class RenderTemplatedICL(RenderAutoFormatTemplate):
     size_limiter: Artifact = None
     input_output_separator: str = "\n"
     demo_separator: str = "\n\n"
+    system_prompt: str = None
 
     def render(self, instance: Dict[str, object]) -> Dict[str, object]:
         demos = instance.pop(self.demos_field, [])
@@ -142,6 +143,9 @@ class RenderTemplatedICL(RenderAutoFormatTemplate):
 
         source += input_str
 
+        if self.system_prompt is not None:
+            source = self.system_prompt.format(source)
+
         return {
             **example,
             "source": source,
@@ -151,7 +155,7 @@ class RenderTemplatedICL(RenderAutoFormatTemplate):
 class InputOutputTemplate(Template):
     input_format: str = None
     output_format: str = None
-    postprocessors: List[str] = field(default_factory=lambda: ["processors.to_string"])
+    postprocessors: List[str] = field(default_factory=lambda: ["processors.to_string_stripped"])
 
     def process_template(self, template: str, data: Dict[str, object]) -> str:
         data = {k: ", ".join(v) if isinstance(v, list) else v for k, v in data.items()}