logo
Browse Source

Update accelerate

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

6
s_bert.py

@ -21,7 +21,11 @@ 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