diff --git a/auto_transformers.py b/auto_transformers.py
index b4055ce..1a06d98 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -25,7 +25,11 @@ from transformers import AutoTokenizer, AutoConfig, AutoModel, AutoModelForMaske
 
 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