Merge pull request #4634 from voxel51/bug/4587

Fix bug when loading HF transformers from zoo
voxel51 · Aug 8, 2024 · 6e3f77c · 6e3f77c
2 parents 8b3b841 + 3aa68f3
commit 6e3f77c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/fiftyone/utils/transformers.py b/fiftyone/utils/transformers.py
@@ -452,7 +452,7 @@ def __init__(self, config):
         self.config = config
         self.model = self._load_model(config)
         self.device = (
-            "cuda" if next(config.model.parameters()).is_cuda else "cpu"
+            "cuda" if next(self.model.parameters()).is_cuda else "cpu"
         )
         self.image_processor = self._load_image_processor()
 
@@ -499,7 +499,7 @@ def __init__(self, config):
         self.classes = config.classes
         self.model = self._load_model(config)
         self.device = (
-            "cuda" if next(config.model.parameters()).is_cuda else "cpu"
+            "cuda" if next(self.model.parameters()).is_cuda else "cpu"
         )
         self.processor = self._load_processor()
         self._text_prompts = None
@@ -748,10 +748,10 @@ def __init__(self, config):
         self.config = config
         self.classes = config.classes
         self.processor = self._load_processor(config)
+        self.model = self._load_model(config)
         self.device = (
-            "cuda" if next(config.model.parameters()).is_cuda else "cpu"
+            "cuda" if next(self.model.parameters()).is_cuda else "cpu"
         )
-        self.model = self._load_model(config)
         self._text_prompts = None
 
     def _load_processor(self, config):