zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-25 02:20:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support Pixal3d and TRELLIS2 (CORE-278) (CORE-199) (CORE-236) (CORE-312) (#14718)

Browse Source
This commit is contained in:
Jukka Seppänen
2026-08-22 03:32:25 +03:00
committed by GitHub
parent 28864ca8fe
commit 0e65cb9071
28 changed files with 12397 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests-unit/comfy_test/model_detection_test.py
View File

@@ -289,6 +289,18 @@ class TestModelDetection:
del sd["double_blocks.0.attn.img_attn_q_norm.weight"]
assert detect_unet_config(sd, "") is None
def test_pixal3d_projection_detection_ignores_packed_weight_width(self):
sd = {
"img2shape.t_embedder.mlp.0.weight": torch.empty(8, 8, device="meta"),
"img2shape.blocks.0.cross_attn.proj_linear.weight": torch.empty(8, 1024, device="meta"),
"structure_model.blocks.0.cross_attn.proj_linear.weight": torch.empty(8, 512, device="meta"),
}
unet_config = detect_unet_config(sd, "")
assert unet_config["proj_in_channels_shape"] == 2048
assert unet_config["proj_in_channels_structure"] == 1024
def test_unet_config_and_required_keys_combination_is_unique(self):
"""Each model in the registry must have a unique combination of
``unet_config`` and ``required_keys``. If two models share the same