SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix auto (#15706)

Browse Source
This commit is contained in:
Lysandre Debut
2022-02-18 08:50:23 -05:00
committed by GitHub
parent d5083c333f
commit 83f45cd656
1 changed files with 14 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
tests/test_modeling_tf_auto.py
View File

@@ -85,35 +85,25 @@ if is_tf_available():
class TFAutoModelTest(unittest.TestCase): class TFAutoModelTest(unittest.TestCase):
@slow @slow
def test_model_from_pretrained(self): def test_model_from_pretrained(self):
import h5py model_name = "bert-base-cased"
config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig)
self.assertTrue(h5py.version.hdf5_version.startswith("1.10")) model = TFAutoModel.from_pretrained(model_name)
self.assertIsNotNone(model)
# for model_name in TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: self.assertIsInstance(model, TFBertModel)
for model_name in ["bert-base-uncased"]:
config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig)
model = TFAutoModel.from_pretrained(model_name)
self.assertIsNotNone(model)
self.assertIsInstance(model, TFBertModel)
@slow @slow
def test_model_for_pretraining_from_pretrained(self): def test_model_for_pretraining_from_pretrained(self):
import h5py model_name = "bert-base-cased"
config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig)
self.assertTrue(h5py.version.hdf5_version.startswith("1.10")) model = TFAutoModelForPreTraining.from_pretrained(model_name)
self.assertIsNotNone(model)
# for model_name in TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST[:1]: self.assertIsInstance(model, TFBertForPreTraining)
for model_name in ["bert-base-uncased"]:
config = AutoConfig.from_pretrained(model_name)
self.assertIsNotNone(config)
self.assertIsInstance(config, BertConfig)
model = TFAutoModelForPreTraining.from_pretrained(model_name)
self.assertIsNotNone(model)
self.assertIsInstance(model, TFBertForPreTraining)
@slow @slow
def test_model_for_causal_lm(self): def test_model_for_causal_lm(self):