talkad commited on
Commit
3130c7f
·
1 Parent(s): fc25784

Explicit congif initialization

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -8,9 +8,14 @@ import json
8
  deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
9
  deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args, use_cuda=False)
10
 
11
- pragformer = transformers.AutoModel.from_pretrained("Pragformer/PragFormer", revision="main", trust_remote_code=True)
12
- pragformer_private = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_private", revision="main", trust_remote_code=True)
13
- pragformer_reduction = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_reduction", revision="main", trust_remote_code=True)
 
 
 
 
 
14
 
15
 
16
  #Event Listeners
 
8
  deep_scc_model_args = ClassificationArgs(num_train_epochs=10,max_seq_length=300,use_multiprocessing=False)
9
  deep_scc_model = ClassificationModel("roberta", "NTUYG/DeepSCC-RoBERTa", num_labels=19, args=deep_scc_model_args, use_cuda=False)
10
 
11
+ pragformer_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer", trust_remote_code=True)
12
+ pragformer = transformers.AutoModel.from_pretrained("Pragformer/PragFormer", config=pragformer_config, trust_remote_code=True)
13
+
14
+ private_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer_private", trust_remote_code=True)
15
+ pragformer_private = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_private", config=private_config, trust_remote_code=True)
16
+
17
+ reduction_config = transformers.AutoConfig.from_pretrained("Pragformer/PragFormer_reduction", trust_remote_code=True)
18
+ pragformer_reduction = transformers.AutoModel.from_pretrained("Pragformer/PragFormer_reduction", config=reduction_config, trust_remote_code=True)
19
 
20
 
21
  #Event Listeners