diff --git a/src/anomalib/models/image/vlm_ad/backends/huggingface.py b/src/anomalib/models/image/vlm_ad/backends/huggingface.py index bfa65555e3..c2bf47244f 100644 --- a/src/anomalib/models/image/vlm_ad/backends/huggingface.py +++ b/src/anomalib/models/image/vlm_ad/backends/huggingface.py @@ -111,7 +111,15 @@ def processor(self) -> "ProcessorMixin": if transformers is None: msg = "transformers is not installed." raise ValueError(msg) - self._processor = transformers.LlavaNextProcessor.from_pretrained(self.model_name) + loaded_processor = transformers.LlavaNextProcessor.from_pretrained( + self.model_name, + revision="main", + ) + if isinstance(loaded_processor, tuple): + self._processor = loaded_processor[0] + else: + self._processor = loaded_processor + assert self._processor is not None return self._processor @property @@ -128,7 +136,15 @@ def model(self) -> "PreTrainedModel": if transformers is None: msg = "transformers is not installed." raise ValueError(msg) - self._model = transformers.LlavaNextForConditionalGeneration.from_pretrained(self.model_name) + loaded_model = transformers.LlavaNextForConditionalGeneration.from_pretrained( + self.model_name, + revision="main", + ) + if isinstance(loaded_model, tuple): + self._model = loaded_model[0] + else: + self._model = loaded_model + assert self._model is not None return self._model @staticmethod