revert

hawkeye217 · Oct 9, 2024 · beae4b9 · beae4b9
1 parent dd6b0c9
commit beae4b9
Showing 1 changed file with 13 additions and 19 deletions.
diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py
@@ -119,27 +119,21 @@ def _load_model_and_tokenizer(self):
             )
 
     def _load_tokenizer(self):
-        if not self.tokenizer:
-            tokenizer_path = os.path.join(
-                f"{MODEL_CACHE_DIR}/{self.model_name}/tokenizer"
-            )
-            self.tokenizer = AutoTokenizer.from_pretrained(
-                self.model_name,
-                cache_dir=tokenizer_path,
-                trust_remote_code=True,
-                clean_up_tokenization_spaces=True,
-            )
+        tokenizer_path = os.path.join(f"{MODEL_CACHE_DIR}/{self.model_name}/tokenizer")
+        return AutoTokenizer.from_pretrained(
+            self.model_name,
+            cache_dir=tokenizer_path,
+            trust_remote_code=True,
+            clean_up_tokenization_spaces=True,
+        )
 
     def _load_feature_extractor(self):
-        if not self.feature_extractor:
-            feature_extractor_path = os.path.join(
-                f"{MODEL_CACHE_DIR}/{self.model_name}/feature_extractor"
-            )
-            self.feature_extractor = AutoFeatureExtractor.from_pretrained(
-                self.model_name,
-                trust_remote_code=True,
-                cache_dir=feature_extractor_path,
-            )
+        feature_extractor_path = os.path.join(
+            f"{MODEL_CACHE_DIR}/{self.model_name}/feature_extractor"
+        )
+        return AutoFeatureExtractor.from_pretrained(
+            self.model_name, trust_remote_code=True, cache_dir=feature_extractor_path
+        )
 
     def _load_model(self, path: str, providers: List[str]):
         if os.path.exists(path):