nawhgnuj commited on
Commit
778b2e1
·
verified ·
1 Parent(s): 65e6974

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -76,11 +76,10 @@ def stream_chat(
76
 
77
  Importantly, always respond to and rebut the previous speaker's points in Trump's style. Keep responses concise and avoid unnecessary repetition."""
78
 
79
- temperature = 0.15
80
- max_new_tokens = 300
81
- top_p = 0.9
82
- top_k = 40
83
- repetition_penalty = 1.5
84
 
85
  conversation = [
86
  {"role": "system", "content": system_prompt}
@@ -108,7 +107,6 @@ Importantly, always respond to and rebut the previous speaker's points in Trump'
108
  pad_token_id=tokenizer.pad_token_id,
109
  eos_token_id=tokenizer.eos_token_id,
110
  streamer=streamer,
111
- repetition_penalty=repetition_penalty,
112
  )
113
 
114
  with torch.no_grad():
 
76
 
77
  Importantly, always respond to and rebut the previous speaker's points in Trump's style. Keep responses concise and avoid unnecessary repetition."""
78
 
79
+ temperature = 0.8
80
+ max_new_tokens = 1024
81
+ top_p = 1.0
82
+ top_k = 20
 
83
 
84
  conversation = [
85
  {"role": "system", "content": system_prompt}
 
107
  pad_token_id=tokenizer.pad_token_id,
108
  eos_token_id=tokenizer.eos_token_id,
109
  streamer=streamer,
 
110
  )
111
 
112
  with torch.no_grad():