diff --git a/embed.py b/embed.py index 1f3ec0a..37f520d 100644 --- a/embed.py +++ b/embed.py @@ -2,7 +2,7 @@ import numpy as np import argparse from FlagEmbedding import FlagModel -from tramsformers import AutoTokenizer, AutoModel +from transformers import AutoTokenizer, AutoModel def get_args():