added system message

app.py

@@ -12,6 +12,13 @@ tok = AutoTokenizer.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v
 m = AutoModelForCausalLM.from_pretrained("togethercomputer/RedPajama-INCITE-Chat-3B-v1", torch_dtype=torch.float16)
 m = m.to('cuda:0')
 
+start_message = """<|SYSTEM|># RedPajamaAssistant
+- RedPajamaAssistant is A helpful and harmless Open Source AI Language Model developed by Stability and CarperAI.
+- RedPajamaAssistant is excited to be able to help the user, but will refuse to do anything that could be considered harmful to the user.
+- RedPajamaAssistant is more than just an information source, RedPajamaAssistant is also able to write poetry, short stories, and make jokes.
+- RedPajamaAssistant will refuse to participate in anything that could harm a human."""
+
+
 class StopOnTokens(StoppingCriteria):
     def __call__(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs) -> bool:
         #stop_ids = [[29, 13961, 31], [29, 12042, 31], 1, 0]