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

Merge pull request #2 from JMW-dX-LDN/ipex-clip-vision-support

Browse Source 
Force clip vision to use cpu for IPEX
This commit is contained in:
Kel Phillipson 2024-08-16 11:05:52 -08:00 committed by GitHub
commit 4a015499df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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.image_size = config.get("image_size", 224)
self.load_device = comfy.model_management.text_encoder_device() 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() offload_device = comfy.model_management.text_encoder_offload_device()
self.dtype = comfy.model_management.text_encoder_dtype(self.load_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) 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): def load_model_gpu(model):
if model.load_device == 'cpu':
return 'cpu'
return load_models_gpu([model]) return load_models_gpu([model])
def loaded_models(only_currently_used=False): def loaded_models(only_currently_used=False):