Omnibus commited on
Commit
369dc1f
1 Parent(s): e9b47ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -25,17 +25,19 @@ def format_prompt(message, history):
25
 
26
 
27
  def chat_inf(system_prompt,prompt,history,client_choice):
28
- client=clients[int(client_choice)-1]
 
29
  if not history:
30
  history = []
31
  hist_len=0
32
  if history:
33
  hist_len=len(history)
34
  print(hist_len)
 
35
  seed = random.randint(1,1111111111111111)
36
  generate_kwargs = dict(
37
  temperature=0.9,
38
- max_new_tokens=10480,
39
  top_p=0.95,
40
  repetition_penalty=1.0,
41
  do_sample=True,
 
25
 
26
 
27
  def chat_inf(system_prompt,prompt,history,client_choice):
28
+ #token max=8192
29
+ client=clients[int(client_choice)]
30
  if not history:
31
  history = []
32
  hist_len=0
33
  if history:
34
  hist_len=len(history)
35
  print(hist_len)
36
+
37
  seed = random.randint(1,1111111111111111)
38
  generate_kwargs = dict(
39
  temperature=0.9,
40
+ max_new_tokens=6000,
41
  top_p=0.95,
42
  repetition_penalty=1.0,
43
  do_sample=True,