From 8d3acfef128af0d08c41b9c4b224d41db20b85d9 Mon Sep 17 00:00:00 2001
From: xujinling <jinling.xu@zilliz.com>
Date: Tue, 14 Jun 2022 19:33:49 +0800
Subject: [PATCH] self

Signed-off-by: xujinling <jinling.xu@zilliz.com>
---
 frozen_in_time.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/frozen_in_time.py b/frozen_in_time.py
index 855328b..35afa91 100644
--- a/frozen_in_time.py
+++ b/frozen_in_time.py
@@ -67,7 +67,7 @@ class FrozenInTime(NNOperator):
                                                  video_is_load_pretrained=False,
                                                  video_model_type='SpaceTimeTransformer',
                                                  text_is_load_pretrained=False,
-                                                 device=device)
+                                                 device=self.device)
 
         self.tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased', TOKENIZERS_PARALLELISM=False)
         self.transform_cfgs = get_configs(