minilm_test.py

from transformers import AutoTokenizer, AutoModel
  
tokenizer = AutoTokenizer.from_pretrained("BM-K/KoMiniLM-68M") # 23M model
model = AutoModel.from_pretrained("BM-K/KoMiniLM-68M")

inputs = tokenizer("안녕 세상아!", return_tensors="pt")
outputs = model(**inputs)

print(outputs)