From 584696dba19f4a547790a4c40608de78cce8b5ec Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Thu, 30 Mar 2023 11:01:48 +0800
Subject: [PATCH] Update accelerate

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 s_bert.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

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