broadfield commited on
Commit
5baa862
·
verified ·
1 Parent(s): 4bb59b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,12 +19,12 @@ def format_prompt(message, history):
19
  return prompt
20
 
21
  def generate(prompt,history,max_new_tokens,seed):
22
- #seed = random.randint(1,9999999999999)
23
  print(seed)
24
  system_prompt = prompts.ASSISTANT
25
  generate_kwargs = dict(
26
  temperature=0.9,
27
- max_new_tokens=max_new_tokens,
28
  top_p=0.95,
29
  repetition_penalty=1.0,
30
  do_sample=True,
 
19
  return prompt
20
 
21
  def generate(prompt,history,max_new_tokens,seed):
22
+ seed = random.randint(1,9999999999999)
23
  print(seed)
24
  system_prompt = prompts.ASSISTANT
25
  generate_kwargs = dict(
26
  temperature=0.9,
27
+ max_new_tokens=512,
28
  top_p=0.95,
29
  repetition_penalty=1.0,
30
  do_sample=True,