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):
self.device = self._get_device()
model.to(self.device)
return tokenizer, processor, model
def _get_device(self) -> str:
if self.device:
device = self.device
......
......@@ -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
......
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