bobpopboom commited on
Commit
c6784b6
·
verified ·
1 Parent(s): 3c7c10f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,7 +23,7 @@ try:
23
  tokenizer=tokenizer,
24
  torch_dtype=torch.float16,
25
  num_return_sequences=1,
26
- do_sample=True,
27
  truncation=True,
28
  max_new_tokens=128
29
  )
@@ -44,7 +44,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
44
  max_new_tokens=max_tokens,
45
  temperature=temperature,
46
  top_p=top_p,
47
- do_sample=True,
48
  pad_token_id=tokenizer.eos_token_id
49
  )[0]["generated_text"]
50
 
 
23
  tokenizer=tokenizer,
24
  torch_dtype=torch.float16,
25
  num_return_sequences=1,
26
+ do_sample=False,
27
  truncation=True,
28
  max_new_tokens=128
29
  )
 
44
  max_new_tokens=max_tokens,
45
  temperature=temperature,
46
  top_p=top_p,
47
+ do_sample=False,
48
  pad_token_id=tokenizer.eos_token_id
49
  )[0]["generated_text"]
50