Fix systag <<SYS>> glitch in some prompts
This commit is contained in:
parent
e9c00df56f
commit
dc7258087d
|
@ -424,9 +424,10 @@ class Interpreter:
|
||||||
print(f"**Removed codeblock**") # TODO: Could add preview of code removed here.
|
print(f"**Removed codeblock**") # TODO: Could add preview of code removed here.
|
||||||
|
|
||||||
def systag(self, beg):
|
def systag(self, beg):
|
||||||
if "mistral" in self.model.lower():
|
lowermodel = self.model.lower()
|
||||||
|
if "llama" in lowermodel:
|
||||||
|
return "[INST]<<SYS>>" if beg else "<</SYS>>[/INST]"
|
||||||
return "[INST]" if beg else "[/INST]\n"
|
return "[INST]" if beg else "[/INST]\n"
|
||||||
return "<<SYS>>" if beg else "<</SYS>>"
|
|
||||||
|
|
||||||
def chat(self, message=None, return_messages=False):
|
def chat(self, message=None, return_messages=False):
|
||||||
global Ginterrupted
|
global Ginterrupted
|
||||||
|
|
Loading…
Reference in New Issue