Nick088 commited on
Commit
2daf6dd
·
verified ·
1 Parent(s): be9466d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -31,6 +31,8 @@ def generate(your_prompt, max_new_tokens, repetition_penalty, temperature, model
31
  torch.manual_seed(seed)
32
  else:
33
  torch.manual_seed(seed)
 
 
34
 
35
  outputs = model.generate(
36
  input_ids,
 
31
  torch.manual_seed(seed)
32
  else:
33
  torch.manual_seed(seed)
34
+
35
+ print(f"seed has been set to {seed}")
36
 
37
  outputs = model.generate(
38
  input_ids,