mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix lowvram (#564)

Browse Source
This commit is contained in:
lllyasviel 2023-10-08 01:15:54 -07:00 committed by GitHub
parent 51cdc5e53a
commit e83828b7b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fooocus_extras/ip_adapter.py
View File

@ -179,7 +179,7 @@ def preprocess(img):
global ip_unconds global ip_unconds
inputs = clip_vision.processor(images=img, return_tensors="pt") inputs = clip_vision.processor(images=img, return_tensors="pt")
comfy.model_management.load_models_gpu([clip_vision.patcher, image_proj_model, ip_layers]) comfy.model_management.load_model_gpu(clip_vision.patcher)
pixel_values = inputs['pixel_values'].to(clip_vision.load_device) pixel_values = inputs['pixel_values'].to(clip_vision.load_device)
if clip_vision.dtype != torch.float32: if clip_vision.dtype != torch.float32:
@ -195,11 +195,15 @@ def preprocess(img):
else: else:
cond = outputs.image_embeds.to(ip_adapter.dtype) cond = outputs.image_embeds.to(ip_adapter.dtype)
comfy.model_management.load_model_gpu(image_proj_model)
cond = image_proj_model.model(cond).to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
comfy.model_management.load_model_gpu(ip_layers)
if ip_unconds is None: if ip_unconds is None:
uncond = ip_negative.to(device=ip_adapter.load_device, dtype=ip_adapter.dtype) uncond = ip_negative.to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_unconds = [m(uncond).cpu() for m in ip_layers.model.to_kvs] ip_unconds = [m(uncond).cpu() for m in ip_layers.model.to_kvs]
cond = image_proj_model.model(cond).to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
ip_conds = [m(cond).cpu() for m in ip_layers.model.to_kvs] ip_conds = [m(cond).cpu() for m in ip_layers.model.to_kvs]
return ip_conds return ip_conds

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.1.3' version = '2.1.4'