From 419f1d761a426d2cc9dcf47679a0f9b3e1fcb8e8 Mon Sep 17 00:00:00 2001 From: Vladimir Blagojevic Date: Wed, 7 Sep 2022 10:19:27 +0200 Subject: [PATCH] Add 15 min timeout for downloading cached HF models --- .github/workflows/tests.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index cfb32fa153..93a33b8353 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -643,11 +643,13 @@ jobs: with: path: ~/.cache/huggingface/transformers/ key: hf-models + env: + SEGMENT_DOWNLOAD_TIMEOUT_MIN: 15 - name: Download models if: steps.cache-hf-models.outputs.cache-hit != 'true' run: | - python -c "from transformers import AutoModel;[AutoModel.from_pretrained(model_name) for model_name in ['vblagoje/bart_lfqa','yjernite/bart_eli5', 'google/pegasus-xsum', 'vblagoje/dpr-ctx_encoder-single-lfqa-wiki', 'vblagoje/dpr-question_encoder-single-lfqa-wiki', 'facebook/dpr-question_encoder-single-nq-base', 'facebook/dpr-ctx_encoder-single-nq-base', 'yjernite/retribert-base-uncased']]" + python -c "from transformers import AutoModel;[AutoModel.from_pretrained(model_name) for model_name in ['vblagoje/bart_lfqa','yjernite/bart_eli5', 'google/pegasus-xsum', 'vblagoje/dpr-ctx_encoder-single-lfqa-wiki', 'vblagoje/dpr-question_encoder-single-lfqa-wiki', 'facebook/dpr-question_encoder-single-nq-base', 'facebook/dpr-ctx_encoder-single-nq-base']]" - name: Run Elasticsearch