Browse Source
Update accelerate
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
6 additions and
2 deletions
-
s_bert.py
|
@ -21,7 +21,11 @@ import torch |
|
|
from sentence_transformers import SentenceTransformer |
|
|
from sentence_transformers import SentenceTransformer |
|
|
|
|
|
|
|
|
from towhee.operator import NNOperator |
|
|
from towhee.operator import NNOperator |
|
|
# from towhee.dc2 import accelerate |
|
|
|
|
|
|
|
|
try: |
|
|
|
|
|
from towhee import accelerate |
|
|
|
|
|
except: |
|
|
|
|
|
def accelerate(func): |
|
|
|
|
|
return func |
|
|
|
|
|
|
|
|
import os |
|
|
import os |
|
|
import warnings |
|
|
import warnings |
|
|