From cb8fb811d79eb4c3d9a0c9f2497f73647c2459fd Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Fri, 10 Feb 2023 18:20:50 +0800
Subject: [PATCH] Update device

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 nn_fingerprint.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nn_fingerprint.py b/nn_fingerprint.py
index e7f8bb7..44e8dc9 100644
--- a/nn_fingerprint.py
+++ b/nn_fingerprint.py
@@ -69,7 +69,7 @@ class Model:
     def __call__(self, *args, **kwargs):
         new_args = []
         new_kwargs = {}
-        for x in new_args:
+        for x in args:
             new_args.append(x.to(self.device))
         for k, v in kwargs.items():
             new_kwargs[k] = v.to(self.device)