xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 05:06:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Force clip vision to use cpu for IPEX

Browse Source
This commit is contained in:
Kel Phillipson 2024-08-16 12:03:23 -07:00
parent 2837938551
commit 818ac32a80
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/clip_vision.py
View File

@ -36,6 +36,8 @@ class ClipVisionModel():
self.image_size = config.get("image_size", 224)
self.load_device = comfy.model_management.text_encoder_device()
# Force CPU for clip vision to work on IPEX
self.load_device = 'cpu'
offload_device = comfy.model_management.text_encoder_offload_device()
self.dtype = comfy.model_management.text_encoder_dtype(self.load_device)
self.model = comfy.clip_model.CLIPVisionModelProjection(config, self.dtype, offload_device, comfy.ops.manual_cast)

2
comfy/model_management.py
View File

@ -540,6 +540,8 @@ def load_models_gpu(models, memory_required=0, force_patch_weights=False, minimu
def load_model_gpu(model):
if model.load_device == 'cpu':
return 'cpu'
return load_models_gpu([model])
def loaded_models(only_currently_used=False):