Repair the transformers corpus

models/transformers/albert.py

@@ -1,9 +1,10 @@
-
 # labels: name::albert author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import AlbertModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/auto.py

@@ -1,9 +1,10 @@
-
 # labels: name::auto author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import AutoModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bart.py

@@ -1,9 +1,10 @@
-
 # labels: name::bart author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BartModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bert.py

@@ -1,9 +1,10 @@
-
 # labels: name::bert author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BertModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bertlmhead.py

@@ -1,9 +1,10 @@
-
 # labels: name::bertlmhead author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BertLMHeadModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bigbird.py

@@ -1,9 +1,10 @@
-
 # labels: name::bigbird author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BigBirdModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bigbirdpegasus.py

@@ -1,9 +1,10 @@
-
 # labels: name::bigbirdpegasus author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BigBirdPegasusModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/biogpt.py

@@ -1,9 +1,10 @@
-
 # labels: name::biogpt author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BioGptModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/blenderbot.py

@@ -1,9 +1,10 @@
-
 # labels: name::blenderbot author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BlenderbotModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -29,4 +30,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/blenderbotsmall.py

@@ -1,9 +1,10 @@
-
 # labels: name::blenderbotsmall author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BlenderbotSmallModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -29,4 +30,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/bloom.py

@@ -1,9 +1,10 @@
-
 # labels: name::bloom author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import BloomModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -25,4 +26,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/camembert.py

@@ -1,9 +1,10 @@
-
 # labels: name::camembert author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import CamembertModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/canine.py

@@ -1,9 +1,10 @@
-
 # labels: name::canine author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import CanineModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/codegen.py

@@ -1,9 +1,10 @@
-
 # labels: name::codegen author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import CodeGenModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/convbert.py

@@ -1,9 +1,10 @@
-
 # labels: name::convbert author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import ConvBertModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/ctrl.py

@@ -1,9 +1,10 @@
-
 # labels: name::ctrl author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import CTRLModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/ctrllmhead.py

@@ -1,9 +1,10 @@
-
 # labels: name::ctrllmhead author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import CTRLLMHeadModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/deberta.py

@@ -1,9 +1,10 @@
-
 # labels: name::deberta author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import DebertaModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/debertav2.py

@@ -1,9 +1,10 @@
-
 # labels: name::debertav2 author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import DebertaV2Model, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/distilbert.py

@@ -1,9 +1,10 @@
-
 # labels: name::distilbert author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import DistilBertModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-

models/transformers/electra.py

@@ -1,9 +1,10 @@
-
 # labels: name::electra author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import ElectraModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -28,4 +29,3 @@
 
 # Call model
 model(**inputs)
-