Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ os.environ['TRANSFORMERS_NO_ADVISORY_WARNINGS'] = '1'
|
|
12 |
def random_state(seed):
|
13 |
torch.manual_seed(seed)
|
14 |
if torch.cuda.is_available():
|
15 |
-
|
16 |
|
17 |
# Initialize Parrot model
|
18 |
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
|
|
|
12 |
def random_state(seed):
|
13 |
torch.manual_seed(seed)
|
14 |
if torch.cuda.is_available():
|
15 |
+
torch.cuda.manual_seed_all(1234)
|
16 |
|
17 |
# Initialize Parrot model
|
18 |
parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5")
|