Update modeling_borealis.py

modeling_borealis.py

@@ -145,7 +145,7 @@ class BorealisForConditionalGeneration(PreTrainedModel, PyTorchModelHubMixin):
         )
         return out.loss, out.logits
 
-    def extract_assistant_content(text: str) -> str:
+    def extract_assistant_content(self, text: str) -> str:
         if "assistant\n" in text:
             return text.split("assistant\n")[-1].strip()
         return text.strip()