xinyun/ComfyUI-Hunyuan3DWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git synced 2025-12-08 20:34:28 +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
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:
image = latents
@ -584,6 +588,7 @@ class HunyuanPaintPipeline(StableDiffusionPipeline):
if not output_type == "latent":
image_list = []
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_list.append(image)
image = torch.cat(image_list, dim=0)