From 16ebb1410b99dce017dc43a3e4bd66a83bd4da92 Mon Sep 17 00:00:00 2001 From: Prithiviraj Damodaran Date: Sat, 8 May 2021 13:16:22 +0000 Subject: [PATCH] Update README.md --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index ca69c5c..3693ee7 100644 --- a/README.md +++ b/README.md @@ -17,13 +17,15 @@ import torch import warnings warnings.filterwarnings("ignore") - +''' +uncomment to get reproducable paraphrase generations def random_state(seed): torch.manual_seed(seed) if torch.cuda.is_available(): torch.cuda.manual_seed_all(seed) 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)