Skip to content

Commit de5437a

Browse files
authored
Merge easy_add_model feature branch (#1309)
* Update to Transformers v4.3.3 (#1266) * use default return_dict in taskmodels and remove hidden state context manager in models. * return hidden states in output of model wrapper * Switch to task model/head factories instead of embedded if-else statements (#1268) * Use jiant transformers model wrapper instead of if-else. Use taskmodel and head factory instead of if-else. * switch to ModelArchitectures enum instead of strings * Refactor get_output_from_encoder() to be member of JiantTaskModel (#1283) * refactor getting output from encoder to be member function of jiant model * switch to explicit encode() in jiant transformers model * fix simple runscript test * update to tokenizer 0.10.1 * Add tests for flat_strip() (#1289) * add flat_strip test * add list to test cases flat_strip * mlm_weights(), feat_spec(), flat_strip() if-else refactors (#1288) * moves remaining if-else statments to jiant model or replaces with model agnostic method * switch from jiant_transformers_model to encoder * fix bug in flat_strip() * Move tokenization logic to central JiantModelTransformers method (#1290) * move model specific tokenization logic to JiantTransformerModels * implement abstract methods for JiantTransformerModels * fix tasks circular import (#1296) * Add DeBERTa (#1295) * Add DeBERTa with sanity test * fix tasks circular import * [WIP] add deberta tests * Revert "fix tasks circular import" This reverts commit f924640. * deberta tests passing with transformers 6472d8 * switch to deberta-v2 * fix get_mlm_weights_dict() for deberta-v2 * update to transformers 4.5.0 * mark deberta test_export as slow * Update test_tokenization_normalization.py * add guide to add a model * fix test_expor_model tests * minor pytest fixes (add num_labels for rte, overnight flag fix) * bugfix for simple api notebook * bugfix for #1310 * bugfix for #1306: simple api notebook path name * squad running * 2nd bugfix for #1310: not all tasks have num_labels property * simple api notebook back to roberta-base * run test matrix for more steps to compare to master * save last/best model test fix Co-authored-by: Jesse Swanson <js11133Wnyu.edu>
1 parent 4d0f6a9 commit de5437a

38 files changed

+1606
-1420
lines changed

conftest.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,15 @@
88
def pytest_addoption(parser):
99
parser.addoption("--runslow", action="store_true", default=False, help="run slow tests")
1010
parser.addoption("--rungpu", action="store_true", default=False, help="run gpu tests")
11-
parser.addoption("--runovernight", action="store_true",
12-
default=False, help="run overnight tests")
11+
parser.addoption(
12+
"--runovernight", action="store_true", default=False, help="run overnight tests"
13+
)
1314

1415

1516
def pytest_configure(config):
1617
config.addinivalue_line("markers", "slow: mark test as slow to run")
1718
config.addinivalue_line("markers", "gpu: mark test as gpu required to run")
19+
config.addinivalue_line("markers", "overnight: mark test as gpu required to run")
1820

1921

2022
def pytest_collection_modifyitems(config, items):

0 commit comments

Comments
 (0)