xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 13:27:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use the CPU for interpolation.

Browse Source
This commit is contained in:
hben35096 2025-05-10 13:47:35 +08:00 committed by GitHub
parent b06b4f67cd
commit 31914c9beb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/clip_vision.py
View File

@ -29,7 +29,12 @@ def clip_preprocess(image, size=224, mean=[0.48145466, 0.4578275, 0.40821073], s
else:
scale_size = (size, size)
image = torch.nn.functional.interpolate(image, size=scale_size, mode="bilinear")
if image.device.type == 'musa':
image = image.cpu()
image = torch.nn.functional.interpolate(image, size=scale_size, mode="bicubic", antialias=True)
image = image.to('musa')
else:
image = torch.nn.functional.interpolate(image, size=scale_size, mode="bicubic", antialias=True)
h = (image.shape[2] - size)//2
w = (image.shape[3] - size)//2
image = image[:,:,h:h+size,w:w+size]