Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Revert "skip free_memory for interrogate"

Browse Source 
This reverts commit 316fe24f65.
Rather add interrogate action to queue than keeping 400MB BLIP loaded in VRAM.
This commit is contained in:
Manuel Schmid
2023-12-31 00:26:59 +01:00
parent 64c1a3dcce
commit 13bb314cfa
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extras/interrogate.py
+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, should_free_memory=False)
model_management.load_model_gpu(self.blip_model)
gpu_image = transforms.Compose([
transforms.ToTensor(),