nawhgnuj commited on
Commit
f6ee02e
·
verified ·
1 Parent(s): e262200

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -76,13 +76,11 @@ def stream_chat(
76
 
77
  Crucially, always respond to and rebut the previous speaker's points in Harris's style. Keep responses concise and impactful."""
78
 
79
- temperature = 0.15
80
- max_new_tokens = 300
81
- top_p = 0.8
82
- top_k = 40
83
- repetition_penalty = 1.5
84
 
85
-
86
  conversation = [
87
  {"role": "system", "content": system_prompt}
88
  ]
@@ -109,7 +107,6 @@ Crucially, always respond to and rebut the previous speaker's points in Harris's
109
  pad_token_id=tokenizer.pad_token_id,
110
  eos_token_id=tokenizer.eos_token_id,
111
  streamer=streamer,
112
- repetition_penalty=repetition_penalty,
113
  )
114
 
115
  with torch.no_grad():
 
76
 
77
  Crucially, always respond to and rebut the previous speaker's points in Harris's style. Keep responses concise and impactful."""
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}
86
  ]
 
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():