remove torchscript

chcwww · chcwww · commit ae94b8bc8877 · 2026-02-12T07:15:50.000Z
diff --git a/libmultilabel/nn/networks/bert_attention.py b/libmultilabel/nn/networks/bert_attention.py
@@ -40,7 +40,7 @@ def __init__(
 
         self.lm = AutoModel.from_pretrained(
             lm_weight,
-            torchscript=True,
+            # 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
`@@ -40,7 +40,7 @@ def __init__(`
`40`	`40`
`41`	`41`	`self.lm = AutoModel.from_pretrained(`
`42`	`42`	`lm_weight,`
`43`		`- torchscript=True,`
	`43`	`+ # torchscript=True,`
`44`	`44`	`hidden_dropout_prob=encoder_hidden_dropout,`
`45`	`45`	`attention_probs_dropout_prob=encoder_attention_dropout,`
`46`	`46`	`)`