Skip to content
Snippets Groups Projects
Unverified Commit 87a6dcbc authored by James Briggs's avatar James Briggs
Browse files

lint

parent 787997a5
No related branches found
No related tags found
No related merge requests found
...@@ -89,7 +89,7 @@ class CLIPEncoder(BaseEncoder): ...@@ -89,7 +89,7 @@ class CLIPEncoder(BaseEncoder):
self.device = self._get_device() self.device = self._get_device()
model.to(self.device) model.to(self.device)
return tokenizer, processor, model return tokenizer, processor, model
def _get_device(self) -> str: def _get_device(self) -> str:
if self.device: if self.device:
device = self.device device = self.device
......
...@@ -65,7 +65,7 @@ class VitEncoder(BaseEncoder): ...@@ -65,7 +65,7 @@ class VitEncoder(BaseEncoder):
model.to(self.device) model.to(self.device)
return processor, model return processor, model
def _get_device(self) -> str: def _get_device(self) -> str:
if self.device: if self.device:
device = self.device device = self.device
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment