Update README.md

This commit is contained in:
Prithiviraj Damodaran 2021-05-08 12:42:49 +00:00 committed by huggingface-web
parent 53fa195f05
commit 3e62bba831
1 changed files with 1 additions and 1 deletions

View File

@ -23,7 +23,7 @@ def set_seed(seed):
if torch.cuda.is_available():
torch.cuda.manual_seed_all(seed)
set_seed(42)
set_seed(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)