Eable acc

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · b176774f3e
1 changed files with 1 additions and 1 deletions
--- a/nn_fingerprint.py
+++ b/nn_fingerprint.py
@ -42,7 +42,7 @@ log = logging.getLogger('nnfp_op')
 log.setLevel(logging.ERROR)
 # @accelerate
@accelerate
 class Model:
    def __init__(self, params, device='cpu', model_path=None):
        self.device = device