Browse Source

Minor bug fix

pull/46/head
bolshoytoster 2 years ago
parent
commit
68460a19eb
  1. 1
      clip_interrogator/clip_interrogator.py

1
clip_interrogator/clip_interrogator.py

@ -440,6 +440,7 @@ class Interrogator:
if self.blip_loaded: # Move CLIP to intended device if self.blip_loaded: # Move CLIP to intended device
self.blip_model.to("cpu") self.blip_model.to("cpu")
self.cli_model.to(self.device) self.cli_model.to(self.device)
self.blip_loaded = False
image_features = self.image_to_features(image) image_features = self.image_to_features(image)
flaves = self.flavors.rank( flaves = self.flavors.rank(

Loading…
Cancel
Save