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

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 auto_transformers.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/auto_transformers.py b/auto_transformers.py
index 02422d2..1727b03 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -23,7 +23,11 @@ from transformers import AutoTokenizer, AutoModel, AutoModelForMaskedLM, AutoMod
 
 from towhee.operator import NNOperator
 from towhee import register
-# from towhee.dc2 import accelerate
+try:
+    from towhee import accelerate
+except:
+    def accelerate(func):
+        return func
 
 import warnings
 import logging