@@ -214,16 +214,28 @@ jobs:
214
214
# * https://github.com/pytorch/pytorch/issues/140965 (aten::_linalg_eigvals)
215
215
# * https://github.com/huggingface/transformers/issues/36267 (marian tests)
216
216
- test_case : ' tests_models_0'
217
- cmd : ' tests/models --num-shards 4 --shard-id 0 --ignore=tests/models/marian/test_modeling_marian.py'
217
+ cmd : ' tests/models --num-shards 8 --shard-id 0 --ignore=tests/models/marian/test_modeling_marian.py'
218
218
filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
219
219
- test_case : ' tests_models_1'
220
- cmd : ' tests/models --num-shards 4 --shard-id 1 --ignore=tests/models/marian/test_modeling_marian.py'
220
+ cmd : ' tests/models --num-shards 8 --shard-id 1 --ignore=tests/models/marian/test_modeling_marian.py'
221
221
filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
222
222
- test_case : ' tests_models_2'
223
- cmd : ' tests/models --num-shards 4 --shard-id 2 --ignore=tests/models/marian/test_modeling_marian.py'
223
+ cmd : ' tests/models --num-shards 8 --shard-id 2 --ignore=tests/models/marian/test_modeling_marian.py'
224
224
filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
225
225
- test_case : ' tests_models_3'
226
- cmd : ' tests/models --num-shards 4 --shard-id 3 --ignore=tests/models/marian/test_modeling_marian.py'
226
+ cmd : ' tests/models --num-shards 8 --shard-id 3 --ignore=tests/models/marian/test_modeling_marian.py'
227
+ filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
228
+ - test_case : ' tests_models_4'
229
+ cmd : ' tests/models --num-shards 8 --shard-id 4 --ignore=tests/models/marian/test_modeling_marian.py'
230
+ filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
231
+ - test_case : ' tests_models_5'
232
+ cmd : ' tests/models --num-shards 8 --shard-id 5 --ignore=tests/models/marian/test_modeling_marian.py'
233
+ filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
234
+ - test_case : ' tests_models_6'
235
+ cmd : ' tests/models --num-shards 8 --shard-id 6 --ignore=tests/models/marian/test_modeling_marian.py'
236
+ filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
237
+ - test_case : ' tests_models_7'
238
+ cmd : ' tests/models --num-shards 8 --shard-id 7 --ignore=tests/models/marian/test_modeling_marian.py'
227
239
filter : ' not test_resize_embeddings_untied and not test_resize_tokens_embeddings'
228
240
# Excluding tests due to:
229
241
# * Some ray tests hang, reason unknown
0 commit comments