When blip_offload enabled keep BLIP model on CPU to start.

2 years ago · bf7404d7fa
1 changed files with 2 additions and 1 deletions
--- a/clip_interrogator/clip_interrogator.py
+++ b/clip_interrogator/clip_interrogator.py
@ -89,6 +89,7 @@ class Interrogator():
                med_config=med_config
            )
            blip_model.eval()
            if not self.config.blip_offload:
                blip_model = blip_model.to(config.device)
            self.blip_model = blip_model
        else: