xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-09 21:04:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update conditioner.py

Browse Source
This commit is contained in:
kijai 2025-03-19 08:31:35 +02:00
parent e7f7877026
commit d7a9ca06bd
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
hy3dgen/shapegen/models/conditioner.py
View File

@ -137,10 +137,12 @@ class ImageEncoder(nn.Module):
for view_tag, view_image in view_dict.items(): for view_tag, view_image in view_dict.items():
if view_image is not None: if view_image is not None:
if view_image.shape[1] == 4: if view_image.shape[1] == 4:
log.info('received image with alpha channel, masking out background...') log.info(f'{view_tag} view image has alpha channel, using it to set background to black')
rgb = view_image[:, :3, :, :] rgb = view_image[:, :3, :, :]
alpha = view_image[:, 3:4, :, :] alpha = view_image[:, 3:4, :, :]
view_image = rgb * alpha view_image = rgb * alpha
else:
log.warning(f"{view_tag} view image has no alpha channel, make sure the background is already black")
view_image = view_image.to(self.model.device, dtype=self.model.dtype) view_image = view_image.to(self.model.device, dtype=self.model.dtype)
if (view_image.shape[2] not in supported_sizes or view_image.shape[3] not in supported_sizes): if (view_image.shape[2] not in supported_sizes or view_image.shape[3] not in supported_sizes):
log.info(f'view_image shape {view_image.shape} not supported. Resizing to 518x518') log.info(f'view_image shape {view_image.shape} not supported. Resizing to 518x518')