xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2026-05-29 05:26:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update pipeline.py

Browse Source
This commit is contained in:
kijai 2025-02-05 01:11:47 +02:00
parent 2846bbf5cc
commit 9d8dfb0700
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
hy3dgen/texgen/hunyuanpaint/pipeline.py
View File

@ -224,7 +224,11 @@ class HunyuanPaintPipeline(StableDiffusionPipeline):
).images ).images
if not output_type == "latent": if not output_type == "latent":
image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0] image_list = []
for img in latents:
image = self.vae.decode(img.unsqueeze(0) / self.vae.config.scaling_factor, return_dict=False)[0]
image_list.append(image)
image = torch.cat(image_list, dim=0)
else: else:
image = latents image = latents
@ -584,6 +588,7 @@ class HunyuanPaintPipeline(StableDiffusionPipeline):
if not output_type == "latent": if not output_type == "latent":
image_list = [] image_list = []
for img in latents: for img in latents:
print("decoding image", img.shape)
image = self.vae.decode(img / self.vae.config.scaling_factor, return_dict=False, generator=generator)[0] image = self.vae.decode(img / self.vae.config.scaling_factor, return_dict=False, generator=generator)[0]
image_list.append(image) image_list.append(image)
image = torch.cat(image_list, dim=0) image = torch.cat(image_list, dim=0)