Elijahbodden commited on
Commit
52963d7
Β·
verified Β·
1 Parent(s): 267d0d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,9 +28,9 @@ model = Llama.from_pretrained(
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
 
30
  presets = {
31
- "Default" : [{"role": "user", "content": "good convo, bye"}, {"role": "assistant", "content": "Haha cool ttyl"}],
32
- "Rizz ????" : [{"role": "user", "content": "omg it's so hot when you flirt with me"}, {"role": "assistant", "content": "haha well you're lucky can even string a sentence together, the way you take my breath away 😘"}, {"role": "user", "content": "alright love you, gn!"}, {"role": "assistant", "content": "ttyl babe πŸ’•"}],
33
- "Thinky" : [{"role": "user", "content": "Woah you just totally blew my mind\ngehh now the fermi paradox is going to be bugging me 24/7\nok ttyl"}, {"role": "assistant", "content": "nah our deep convos are always the best, we should talk again soon\nttyl"}],
34
  }
35
 
36
  def custom_lp_logits_processor(ids, logits, lp_start, lp_decay, prompt_tok_len):
 
28
  tokenizer = AutoTokenizer.from_pretrained(model_id)
29
 
30
  presets = {
31
+ "Default" : [{"from": "user", "value": "good convo, bye"}, {"from": "assistant", "value": "Haha cool ttyl"}],
32
+ "Rizz ????" : [{"from": "user", "value": "omg it's so hot when you flirt with me"}, {"from": "assistant", "value": "haha well you're lucky can even string a sentence together, the way you take my breath away 😘"}, {"from": "user", "value": "alright love you, gn!"}, {"from": "assistant", "value": "ttyl babe πŸ’•"}],
33
+ "Thinky" : [{"from": "user", "value": "Woah you just totally blew my mind\ngehh now the fermi paradox is going to be bugging me 24/7\nok ttyl"}, {"from": "assistant", "value": "nah our deep convos are always the best, we should talk again soon\nttyl"}],
34
  }
35
 
36
  def custom_lp_logits_processor(ids, logits, lp_start, lp_decay, prompt_tok_len):