diff --git a/semantic_router/encoders/clip.py b/semantic_router/encoders/clip.py index 67e58e86234e0a9bc739c1e2c2cc5fd352e6e790..a910a999393b79bc39aa532edd9fa153f25d2651 100644 --- a/semantic_router/encoders/clip.py +++ b/semantic_router/encoders/clip.py @@ -89,7 +89,7 @@ class CLIPEncoder(BaseEncoder): self.device = self._get_device() model.to(self.device) return tokenizer, processor, model - + def _get_device(self) -> str: if self.device: device = self.device diff --git a/semantic_router/encoders/vit.py b/semantic_router/encoders/vit.py index 9e64a7a80d339c98ae2b9102efe2913f33aab92b..65de365ac6abca6a941e5e51e6e538f1a8449a34 100644 --- a/semantic_router/encoders/vit.py +++ b/semantic_router/encoders/vit.py @@ -65,7 +65,7 @@ class VitEncoder(BaseEncoder): model.to(self.device) return processor, model - + def _get_device(self) -> str: if self.device: device = self.device