Omnibus commited on
Commit
7e2719d
·
verified ·
1 Parent(s): 06461b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ top_p=0.95
22
  repetition_penalty=1.0
23
 
24
 
25
- def compress_history(history,temperature=temperature,top_p=top_p,repition_penalty=repition_penalty):
26
  formatted_prompt=f"{COMPRESS_HISTORY.format(history=history)}"
27
  generate_kwargs = dict(
28
  temperature=temperature,
@@ -41,7 +41,7 @@ def compress_history(history,temperature=temperature,top_p=top_p,repition_penalt
41
 
42
  MAX_HISTORY=100
43
 
44
- def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,repition_penalty=repition_penalty):
45
  temperature = float(temperature)
46
  if temperature < 1e-2:
47
  temperature = 1e-2
 
22
  repetition_penalty=1.0
23
 
24
 
25
+ def compress_history(history,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
26
  formatted_prompt=f"{COMPRESS_HISTORY.format(history=history)}"
27
  generate_kwargs = dict(
28
  temperature=temperature,
 
41
 
42
  MAX_HISTORY=100
43
 
44
+ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
45
  temperature = float(temperature)
46
  if temperature < 1e-2:
47
  temperature = 1e-2