logo
Browse Source

Update accelerate

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
584696dba1
  1. 8
      s_bert.py

8
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

Loading…
Cancel
Save