diff --git a/clip_interrogator/clip_interrogator.py b/clip_interrogator/clip_interrogator.py index 8da0472..8a92b6e 100644 --- a/clip_interrogator/clip_interrogator.py +++ b/clip_interrogator/clip_interrogator.py @@ -89,7 +89,8 @@ class Interrogator(): med_config=med_config ) blip_model.eval() - blip_model = blip_model.to(config.device) + if not self.config.blip_offload: + blip_model = blip_model.to(config.device) self.blip_model = blip_model else: self.blip_model = config.blip_model