enable acc

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · 77cc6cca77
1 changed files with 1 additions and 1 deletions
--- 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