Update README.md
This commit is contained in:
parent
53fa195f05
commit
3e62bba831
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue