randeom commited on
Commit
4e74161
·
verified ·
1 Parent(s): fd050a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def clean_generated_text(text):
32
  clean_text = re.sub(r'</s>$', '', text).strip()
33
  return clean_text
34
 
35
- def generate_text(prompt, temperature=0.9, max_new_tokens=2512, top_p=0.95, repetition_penalty=1.0):
36
  temperature = max(temperature, 1e-2)
37
  generate_kwargs = dict(
38
  temperature=temperature,
 
32
  clean_text = re.sub(r'</s>$', '', text).strip()
33
  return clean_text
34
 
35
+ def generate_text(prompt, temperature=0.9, max_new_tokens=1512, top_p=0.95, repetition_penalty=1.0):
36
  temperature = max(temperature, 1e-2)
37
  generate_kwargs = dict(
38
  temperature=temperature,