Jiahuita
/

NewsSourceClassification

Text Classification

news-classification

Eval Results (legacy)

Model card Files Files and versions

Jiahuita commited on Dec 5, 2024

Commit

1833f3a

·

1 Parent(s): bb8ddb0

same issue diff place

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -26,7 +26,7 @@ class NewsClassifierConfig(PretrainedConfig):
 class NewsClassifier(PreTrainedModel):
     config_class = NewsClassifierConfig
-    base_model_prefix = "news_classifier"
     def __init__(self, config):
         super().__init__(config)

 class NewsClassifier(PreTrainedModel):
     config_class = NewsClassifierConfig
+    base_model_prefix = "text_classifier"
     def __init__(self, config):
         super().__init__(config)