zmbfeng commited on
Commit
cc59d37
·
1 Parent(s): bee79a3

encode plus to encode

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def create_response(input_str,
47
  print("num_beams=" + str(num_beams))
48
  seed=int(seed)
49
  print("seed" + str(seed))
50
- encoded = tokenizer.encode_plus(input_str + tokenizer.eos_token, return_tensors="pt")
51
  input_ids = encoded["input_ids"]
52
  #attention_mask = encoded["attention_mask"]
53
 
 
47
  print("num_beams=" + str(num_beams))
48
  seed=int(seed)
49
  print("seed" + str(seed))
50
+ encoded = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
51
  input_ids = encoded["input_ids"]
52
  #attention_mask = encoded["attention_mask"]
53