From 87a6dcbc3d91dff87a5d83a79d5e1e00bafa7607 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 25 Feb 2024 15:21:39 +0400
Subject: [PATCH] lint
---
semantic_router/encoders/clip.py | 2 +-
semantic_router/encoders/vit.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/semantic_router/encoders/clip.py b/semantic_router/encoders/clip.py
index 67e58e86..a910a999 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 9e64a7a8..65de365a 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
--
GitLab