diff --git a/s_bert.py b/s_bert.py index 7586c2e..77f5dd1 100644 --- a/s_bert.py +++ b/s_bert.py @@ -21,8 +21,12 @@ import torch from sentence_transformers import SentenceTransformer from towhee.operator import NNOperator -# from towhee.dc2 import accelerate - +try: + from towhee import accelerate +except: + def accelerate(func): + return func + import os import warnings