zmbfeng commited on
Commit
b15dd45
·
verified ·
1 Parent(s): 65fba56

use old token function

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ def create_response_original(input_str,
39
  # num_return_sequences=num_return_sequences)[0])
40
 
41
  input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
42
- output_ids = original_model.generate(input_ids,
43
  do_sample=do_sample,
44
  max_length=100,
45
  temperature=temperature,
 
39
  # num_return_sequences=num_return_sequences)[0])
40
 
41
  input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
42
+ output_ids = original_model.generate(**tokenizer(input_str+tokenizer.eos_token,return_tensors="pt",max_length=200),
43
  do_sample=do_sample,
44
  max_length=100,
45
  temperature=temperature,