Fix some prompt formats

#81
lmg-anon · Jun 29, 2024 · e1d2f36 · e1d2f36
1 parent 759d863
commit e1d2f36
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mikupad.html b/mikupad.html
@@ -4184,13 +4184,13 @@
 		'Mistral': {
 			'sysPre' : '<<SYS>>\\n',
 			'sysSuf' : '<</SYS>>\\n\\n',
-			'instPre': '[INST]',
+			'instPre': '</s>[INST]',
 			'instSuf': '[/INST]',
 		},
 		'Codestral': {
 			'sysPre' : '<<SYS>>\\n',
 			'sysSuf' : '<</SYS>>\\n\\n',
-			'instPre': '[INST]',
+			'instPre': '</s>[INST]',
 			'instSuf': '[/INST]',
 			'fimTemplate': '[SUFFIX]{suffix}[PREFIX]{prefix}'
 		},
@@ -4233,14 +4233,14 @@
 		'Vicuna': {
 			'sysPre' : '',
 			'sysSuf' : '\\n\\n',
-			'instPre': '\\nUSER: ',
+			'instPre': '</s>\\nUSER: ',
 			'instSuf': '\\nASSISTANT: ',
 		},
 		'Gemma': {
 			'sysPre' : '',
 			'sysSuf' : '',
-			'instPre': '\\n<end_of_turn><start_of_turn>user\\n',
-			'instSuf': '\\n<end_of_turn><start_of_turn>model\\n',
+			'instPre': '<end_of_turn>\\n<start_of_turn>user\\n',
+			'instSuf': '<end_of_turn>\\n<start_of_turn>model\\n',
 		},
 	},
 	scrollTop: 0