Commit
•
16ebb14
1
Parent(s):
2bdcc83
Update README.md
Browse files
README.md
CHANGED
|
@@ -17,13 +17,15 @@ import torch
|
|
| 17 |
import warnings
|
| 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)
|
|
|
|
| 17 |
import warnings
|
| 18 |
warnings.filterwarnings("ignore")
|
| 19 |
|
| 20 |
+
'''
|
| 21 |
+
uncomment to get reproducable paraphrase generations
|
| 22 |
def random_state(seed):
|
| 23 |
torch.manual_seed(seed)
|
| 24 |
if torch.cuda.is_available():
|
| 25 |
torch.cuda.manual_seed_all(seed)
|
| 26 |
|
| 27 |
random_state(1234)
|
| 28 |
+
'''
|
| 29 |
|
| 30 |
#Init models (make sure you init ONLY once if you integrate this to your code)
|
| 31 |
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)
|