diff --git a/README.md b/README.md index 6ec8ada..32a544c 100644 --- a/README.md +++ b/README.md @@ -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)