remove "torchscript=True" since it is no longer supported by "AutoModel.from_pretrained()"

chcwww · chcwww · commit 2e9719ff164d · 2026-02-13T11:33:46.000Z
diff --git a/.github/workflows/basic_test.yml b/.github/workflows/basic_test.yml
@@ -1,9 +1,9 @@
 name: Basic Test
 
 on:
-  push:
+  pull_request:
     branches:
-      - fix_wf
+      - master
 
 jobs:
   test:
diff --git a/libmultilabel/nn/networks/bert.py b/libmultilabel/nn/networks/bert.py
@@ -34,7 +34,6 @@ def __init__(
             hidden_dropout_prob=encoder_hidden_dropout,
             attention_probs_dropout_prob=encoder_attention_dropout,
             classifier_dropout=post_encoder_dropout,
-            # torchscript=True,
         )
 
     def forward(self, input):
diff --git a/libmultilabel/nn/networks/bert_attention.py b/libmultilabel/nn/networks/bert_attention.py
@@ -40,7 +40,6 @@ def __init__(
 
         self.lm = AutoModel.from_pretrained(
             lm_weight,
-            # torchscript=True,
             hidden_dropout_prob=encoder_hidden_dropout,
             attention_probs_dropout_prob=encoder_attention_dropout,
         )

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,6 @@ def __init__(`
`34`	`34`	`hidden_dropout_prob=encoder_hidden_dropout,`
`35`	`35`	`attention_probs_dropout_prob=encoder_attention_dropout,`
`36`	`36`	`classifier_dropout=post_encoder_dropout,`
`37`		`- # torchscript=True,`
`38`	`37`	`)`
`39`	`38`
`40`	`39`	`def forward(self, input):`
Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,6 @@ def __init__(`
`40`	`40`
`41`	`41`	`self.lm = AutoModel.from_pretrained(`
`42`	`42`	`lm_weight,`
`43`		`- # torchscript=True,`
`44`	`43`	`hidden_dropout_prob=encoder_hidden_dropout,`
`45`	`44`	`attention_probs_dropout_prob=encoder_attention_dropout,`
`46`	`45`	`)`