Basic support for the ace step 1.5 model. (#12237)

This commit is contained in:
comfyanonymous
2026-02-02 21:06:18 -08:00
committed by GitHub
parent ba5bf3f1a8
commit 3c1a1a2df8
12 changed files with 1566 additions and 8 deletions

View File

@@ -155,6 +155,8 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder):
self.execution_device = options.get("execution_device", self.execution_device)
if isinstance(self.layer, list) or self.layer == "all":
pass
elif isinstance(layer_idx, list):
self.layer = layer_idx
elif layer_idx is None or abs(layer_idx) > self.num_layers:
self.layer = "last"
else: