zmbfeng commited on
Commit
4aa8f13
·
verified ·
1 Parent(s): 004631e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -45,6 +45,7 @@ def create_response(input_str,
45
  if not do_sample:
46
  num_beams = 1
47
  print("num_beams=" + str(num_beams))
 
48
  print("seed" + str(seed))
49
  encoded = tokenizer.encode_plus(input_str + tokenizer.eos_token, return_tensors="pt")
50
  input_ids = encoded["input_ids"]
 
45
  if not do_sample:
46
  num_beams = 1
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"]