From 77cc6cca7792f6a96d2c48f1c8f40b75733ae55c Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Wed, 19 Jul 2023 17:12:01 +0800
Subject: [PATCH] enable acc

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 auto_transformers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/auto_transformers.py b/auto_transformers.py
index 1727b03..5d12fb9 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -56,7 +56,7 @@ def create_model(model_name, checkpoint_path, device):
     return model
 
 
-# @accelerate
+@accelerate
 class Model:
     def __init__(self, model_name, checkpoint_path, device):
         self.device = device