[CI] Retry Hugging Face processor loading (#49908)
Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Andreas Karatzas committed
Jul 30, 2026 at 19:48 UTC
553fcb82d5602c75fb6ab41b6dc3c46f480c1785
1 file changed
+10
-3
tests/conftest.py
+10
-3
@@ -66,6 +66,7 @@ from vllm.multimodal.utils import fetch_image
66
from vllm.outputs import RequestOutput
67
from vllm.platforms import current_platform
68
from vllm.sampling_params import BeamSearchParams
69
+from vllm.transformers_utils.repo_utils import with_retry
70
from vllm.transformers_utils.utils import maybe_model_redirect
71
from vllm.utils.collection_utils import is_list_of
72
from vllm.utils.torch_utils import set_default_torch_num_threads
@@ -543,9 +544,15 @@ class HfRunner:
544
# it will call torch.accelerator.device_count()
545
from transformers import AutoProcessor
546
546
- self.processor = AutoProcessor.from_pretrained(
547
- model_name,
548
- trust_remote_code=trust_remote_code,
547
+ # A concurrent refresh of the shared HF cache can briefly hide
548
+ # processor configuration files. Retry just as model config loading
549
+ # does in vllm.transformers_utils.config.
550
+ self.processor = with_retry(
551
+ lambda: AutoProcessor.from_pretrained(
552
+ model_name,
553
+ trust_remote_code=trust_remote_code,
554
+ ),
555
+ f"Error loading processor for {model_name}",
556
)
557
if skip_tokenizer_init:
558
if self.processor is None: