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 12:43:09 +00:00 committed by huggingface-web
parent 3e62bba831
commit 2bdcc837e3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -18,12 +18,12 @@ import warnings
warnings.filterwarnings("ignore")
def set_seed(seed):
def random_state(seed):
torch.manual_seed(seed)
if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed)
set_seed(1234)
random_state(1234)
#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)