[ci] Load pretrained models into the default (long-lived) cache

There's an inconsistency right now where:
- we load some models into CACHE_DIR
- and some models in the default cache
- and often, in both for the same models

When running the RUN_SLOW tests, this takes a lot of disk space, time, and bandwidth.

I'd rather always use the default cache
This commit is contained in:
Julien Chaumond committed 2020-04-30 22:30:15 -04:00
1 parent 6b410bedfc
commit f54dc3f4d5
27 files changed
+50 -53

No files matched your search

+1 -1
View File
@@ -248,5 +248,5 @@ class DistilBertModelTest(ModelTesterMixin, unittest.TestCase):
# @slow
# def test_model_from_pretrained(self):
# for model_name in list(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_MAP.keys())[:1]:
# model = DistilBertModel.from_pretrained(model_name, cache_dir=CACHE_DIR)
# model = DistilBertModel.from_pretrained(model_name)
# self.assertIsNotNone(model)