wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-28 03:12:31 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

clarify: issue 86 latent-to-pixel resolution mapping

Browse Source
This commit is contained in:
John Pollock 2026-04-19 19:20:14 -05:00
parent d62bbe5fe0
commit 5575e06ff3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/ldm/trellis2/model.py
View File

@ -782,7 +782,9 @@ class Trellis2(nn.Module):
embeds = kwargs.get("embeds") embeds = kwargs.get("embeds")
if embeds is None: if embeds is None:
raise ValueError("Trellis2.forward requires 'embeds' in kwargs") raise ValueError("Trellis2.forward requires 'embeds' in kwargs")
is_1024 = self.img2shape.resolution == 64 # img2shape.resolution is the latent-grid size, not the input pixel size:
# 32 -> 512px path, 64 -> 1024px path.
uses_1024_conditioning = self.img2shape.resolution == 64
coords = transformer_options.get("coords", None) coords = transformer_options.get("coords", None)
mode = transformer_options.get("generation_mode", "structure_generation") mode = transformer_options.get("generation_mode", "structure_generation")
is_512_run = False is_512_run = False
@ -797,7 +799,7 @@ class Trellis2(nn.Module):
mode = "structure_generation" mode = "structure_generation"
not_struct_mode = False not_struct_mode = False
if is_1024 and not_struct_mode and not is_512_run: if uses_1024_conditioning and not_struct_mode and not is_512_run:
context = embeds context = embeds
sigmas = transformer_options.get("sigmas")[0].item() sigmas = transformer_options.get("sigmas")[0].item()