prithivida commited on
Commit
2bdcc83
1 Parent(s): 3e62bba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -18,12 +18,12 @@ import warnings
18
  warnings.filterwarnings("ignore")
19
 
20
 
21
- def set_seed(seed):
22
  torch.manual_seed(seed)
23
  if torch.cuda.is_available():
24
  torch.cuda.manual_seed_all(seed)
25
 
26
- set_seed(1234)
27
 
28
  #Init models (make sure you init ONLY once if you integrate this to your code)
29
  parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)
 
18
  warnings.filterwarnings("ignore")
19
 
20
 
21
+ def random_state(seed):
22
  torch.manual_seed(seed)
23
  if torch.cuda.is_available():
24
  torch.cuda.manual_seed_all(seed)
25
 
26
+ random_state(1234)
27
 
28
  #Init models (make sure you init ONLY once if you integrate this to your code)
29
  parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)