wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-06 11:32:31 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Update clip_vision.py

Browse Source 
fix(clip_vision): update line 59 load_model_gpu(self.patcher) to load_models_gpu([self.patcher]) to match model_management.py (for plural function) and iterable argument requirement
This commit is contained in:
simbo1005 2026-02-03 03:03:26 +02:00 committed by GitHub
parent c05a08ae66
commit 75f6fad091
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/clip_vision.py
View File

@ -56,7 +56,7 @@ class ClipVisionModel():
return self.model.state_dict()
def encode_image(self, image, crop=True):
comfy.model_management.load_model_gpu(self.patcher)
comfy.model_management.load_models_gpu([self.patcher])
if self.model_type == "siglip2_vision_model":
pixel_values = comfy.clip_model.siglip2_preprocess(image.to(self.load_device), size=self.image_size, patch_size=self.config.get("patch_size", 16), num_patches=self.config.get("num_patches", 256), mean=self.image_mean, std=self.image_std, crop=crop).float()
else: