orik-ss commited on
Commit
7905374
·
1 Parent(s): e06b945

Fixed siglip2 issue

Browse files
Files changed (1) hide show
  1. siglip2_onnx_zeroshot.py +1 -1
siglip2_onnx_zeroshot.py CHANGED
@@ -57,7 +57,7 @@ class SigLIP2ONNXClassifier:
57
  self.text_session = _make_session(text_path, device)
58
 
59
  # Processor handles both image preprocessing and tokenization
60
- self.processor = AutoProcessor.from_pretrained(REPO_ID)
61
 
62
  # Map I/O names
63
  self._vision_input_names = [i.name for i in self.vision_session.get_inputs()]
 
57
  self.text_session = _make_session(text_path, device)
58
 
59
  # Processor handles both image preprocessing and tokenization
60
+ self.processor = AutoProcessor.from_pretrained(REPO_ID, use_fast=False)
61
 
62
  # Map I/O names
63
  self._vision_input_names = [i.name for i in self.vision_session.get_inputs()]