P01son commited on
Commit
2e1046a
·
1 Parent(s): 516f098

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ def process(message, history):
21
  input_prompt = f"{input_prompt} ### Instruction:{message.strip()} ### Response:"
22
  inputs = tokenizer(input_prompt, return_tensors="pt").to("cuda:0")
23
  try:
24
- generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample=True, top_k=5, top_p=0.84,
25
- temperature=0.5, repetition_penalty=1.15, eos_token_id=2, bos_token_id=1,
26
  pad_token_id=0)
27
  response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
28
  print('log:', response)
 
21
  input_prompt = f"{input_prompt} ### Instruction:{message.strip()} ### Response:"
22
  inputs = tokenizer(input_prompt, return_tensors="pt").to("cuda:0")
23
  try:
24
+ generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample=True, top_k=20, top_p=0.84,
25
+ temperature=1, repetition_penalty=1.15, eos_token_id=2, bos_token_id=1,
26
  pad_token_id=0)
27
  response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
28
  print('log:', response)