prithivida
/
parrot_paraphraser_on_T5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Prithiviraj Damodaran 2021-05-08 13:16:22 +00:00 committed by huggingface-web
parent 2bdcc837e3
commit 16ebb1410b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -17,13 +17,15 @@ import torch
import warnings import warnings
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
'''
uncomment to get reproducable paraphrase generations
def random_state(seed): def random_state(seed):
torch.manual_seed(seed) torch.manual_seed(seed)
if torch.cuda.is_available(): if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed) torch.cuda.manual_seed_all(seed)
random_state(1234) random_state(1234)
'''
#Init models (make sure you init ONLY once if you integrate this to your code) #Init models (make sure you init ONLY once if you integrate this to your code)
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False) parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)