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

try fix some mps problems

Browse Source
This commit is contained in:
lllyasviel
2023-10-23 20:19:24 -07:00
parent 60cb91c406
commit 10a9f0fc9d
3 changed files with 5 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fooocus_extras/ip_adapter.py
+4 -2
View File
@@ -162,9 +162,11 @@ def preprocess(img):
outputs = clip_vision.model(pixel_values=pixel_values, output_hidden_states=True)
if ip_adapter.plus:
cond = outputs.hidden_states[-2].to(ip_adapter.dtype)
cond = outputs.hidden_states[-2]
else:
cond = outputs.image_embeds.to(ip_adapter.dtype)
cond = outputs.image_embeds
cond = cond.to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)
fcbh.model_management.load_model_gpu(image_proj_model)
cond = image_proj_model.model(cond).to(device=ip_adapter.load_device, dtype=ip_adapter.dtype)