From 082f59f2771ae01e69d2dcb9d09352a01598a5e2 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Mon, 19 Dec 2022 10:38:35 +0800
Subject: [PATCH] Change accelerate to comment since not yet supported in
 latest release

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

diff --git a/auto_transformers.py b/auto_transformers.py
index ff193d5..f1219da 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -22,7 +22,7 @@ from transformers import AutoTokenizer, AutoModel, AutoModelForMaskedLM, AutoMod
 
 from towhee.operator import NNOperator
 from towhee import register
-from towhee.dc2 import accelerate
+# from towhee.dc2 import accelerate
 
 import warnings
 import logging
@@ -35,7 +35,7 @@ log = logging.getLogger('run_op')
 warnings.filterwarnings('ignore')
 
 
-@accelerate
+# @accelerate
 class Model:
     def __init__(self, model_name, device, checkpoint_path):
         try: