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

encode plus to encode fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ 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(input_str + tokenizer.eos_token, return_tensors="pt")
51
- input_ids = encoded["input_ids"]
52
  #attention_mask = encoded["attention_mask"]
53
 
54
  if seed != -1:
 
47
  print("num_beams=" + str(num_beams))
48
  seed=int(seed)
49
  print("seed" + str(seed))
50
+ input_ids = tokenizer.encode(input_str + tokenizer.eos_token, return_tensors="pt")
51
+ #= encoded["input_ids"]
52
  #attention_mask = encoded["attention_mask"]
53
 
54
  if seed != -1: