PLatonG commited on
Commit
3a9a37e
1 Parent(s): a905c3b

Update app.py

Browse files

set torch.manual_seed(42) above generate functiono.

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -123,6 +123,7 @@ def evaluate(
123
  return # early return for stream_output
124
 
125
  # Without streaming
 
126
  with torch.no_grad():
127
  generation_output = model.generate(
128
  input_ids=input_ids,
 
123
  return # early return for stream_output
124
 
125
  # Without streaming
126
+ torch.manual_seed(42)
127
  with torch.no_grad():
128
  generation_output = model.generate(
129
  input_ids=input_ids,