Update splade_encoder.py

splade_encoder.py

@@ -29,10 +29,11 @@ SOFTWARE.
 import torch
 import logging
 import onnxruntime as ort
+from transformers import AutoTokenizer
 from typing import Dict, List, Optional
 from scipy.sparse import csr_array, vstack
 from milvus_model.base import BaseEmbeddingFunction
-from transformers import AutoModelForMaskedLM, AutoTokenizer
+from optimum.onnxruntime import ORTModelForMaskedLM
 from milvus_model.utils import import_transformers, import_scipy, import_torch
 
 import_torch()