skip free_memory for interrogate

This commit is contained in:
Manuel Schmid
2023-12-26 22:34:57 +01:00
parent 7b5bced6c2
commit 316fe24f65
2 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -47,7 +47,7 @@ class Interrogator:
self.blip_model = ModelPatcher(model, load_device=self.load_device, offload_device=self.offload_device)
model_management.load_model_gpu(self.blip_model)
model_management.load_model_gpu(self.blip_model, should_free_memory=False)
gpu_image = transforms.Compose([
transforms.ToTensor(),