pabloce commited on
Commit
2413cb6
·
verified ·
1 Parent(s): f54761b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def respond(
40
  )
41
  history_transformer_format = history + [[message, ""]]
42
 
43
- system_prompt = "<|im_start|>system\nYou are Dolphin, a helpful AI assistant.<|im_end|>"
44
  messages = system_prompt + "".join(["".join(["\n<|im_start|>user\n" + item[0], "<|im_end|>\n<|im_start|>assistant\n" + item[1]]) for item in history_transformer_format])
45
  input_ids = tokenizer([messages], return_tensors="pt").to('cuda')
46
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
40
  )
41
  history_transformer_format = history + [[message, ""]]
42
 
43
+ system_prompt = "<|im_start|>system\nYou are Einstein, a helpful AI assistant.<|im_end|>"
44
  messages = system_prompt + "".join(["".join(["\n<|im_start|>user\n" + item[0], "<|im_end|>\n<|im_start|>assistant\n" + item[1]]) for item in history_transformer_format])
45
  input_ids = tokenizer([messages], return_tensors="pt").to('cuda')
46
  streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)