Merge branch 'master' into fix-doc-build

flairNLP · Aug 14, 2024 · 8f3616d · 8f3616d
2 parents 27308a4 + 091e62b
commit 8f3616d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/models/test_tars_classifier.py b/tests/models/test_tars_classifier.py
@@ -12,7 +12,7 @@ class TestTarsClassifier(BaseModelTest):
     train_label_type = "class"
     model_args = {"task_name": "2_CLASS"}
     training_args = {"mini_batch_size": 1, "max_epochs": 2}
-    pretrained_model = "tars-base"
+    # pretrained_model = "tars-base"  # disabled due to too much space requirements.
 
     @pytest.fixture()
     def corpus(self, tasks_base_path):

diff --git a/tests/models/test_tars_ner.py b/tests/models/test_tars_ner.py
@@ -12,7 +12,7 @@ class TestTarsTagger(BaseModelTest):
     train_label_type = "ner"
     model_args = {"task_name": "2_NER"}
     training_args = {"mini_batch_size": 1, "max_epochs": 2}
-    pretrained_model = "tars-ner"
+    # pretrained_model = "tars-ner"  # disabled due to too much space requirements.
 
     @pytest.fixture()
     def corpus(self, tasks_base_path):