prithivida
commited on
Commit
•
2bdcc83
1
Parent(s):
3e62bba
Update README.md
Browse files
README.md
CHANGED
@@ -18,12 +18,12 @@ import warnings
|
|
18 |
warnings.filterwarnings("ignore")
|
19 |
|
20 |
|
21 |
-
def
|
22 |
torch.manual_seed(seed)
|
23 |
if torch.cuda.is_available():
|
24 |
torch.cuda.manual_seed_all(seed)
|
25 |
|
26 |
-
|
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)
|