Upload simple_latent_passthrough.py

custom_nodes/simple_latent_passthrough.py

@@ -0,0 +1,61 @@
+import torch
+
+class SimpleLatentPassthrough:
+    def __init__(self):
+        pass
+    
+    @classmethod
+    def INPUT_TYPES(s):
+        # ИСПРАВЛЕНО: Ключи приведены к нижнему регистру, теперь входы отобразятся в интерфейсе
+        return {
+            "required": {},
+            "optional": {
+                "latent_1": ("LATENT",),
+                "latent_2": ("LATENT",),
+            }
+        }
+
+    RETURN_TYPES = ("LATENT",)
+    FUNCTION = "route_latent"
+    CATEGORY = "latent"
+
+    def route_latent(self, latent_1=None, latent_2=None):
+        # Сценарий 1: Активен только первый метод (второй отключен)
+        if latent_1 is not None and latent_2 is None:
+            return (latent_1,)
+        
+        # Сценарий 2: Активен только второй метод (первый отключен)
+        if latent_2 is not None and latent_1 is None:
+            return (latent_2,)
+        
+        # Сценарий 3: Если включены обе группы сразу
+        if latent_1 is not None and latent_2 is not None:
+            try:
+                # Так как в ComfyUI нельзя сделать физическое разделение "кто быстрее пришел" (обе ветки считаются параллельно
+                # и приходят в ноду одновременно), лучшим решением для реализации вашего правила "пропустить каждый по очереди"
+                # является объединение их в латентный батч (пакет).
+                # В этом случае KSampler и последующие ноды обработают их последовательно один за другим.
+                samples_1 = latent_1["samples"]
+                samples_2 = latent_2["samples"]
+                
+                # Склеиваем тензоры по размерности батча (dim=0)
+                batched_samples = torch.cat((samples_1, samples_2), dim=0)
+                
+                result = latent_1.copy()
+                result["samples"] = batched_samples
+                return (result,)
+            except Exception as e:
+                # Если латенты разного разрешения и склеить их в батч нельзя, отдаем приоритет первому
+                return (latent_1,)
+        
+        # Сценарий 4: Если оба входа оказались пустыми
+        raise ValueError("SimpleLatentPassthrough: Оба входа пусты. Включите хотя бы одну группу апскейла.")
+
+# Регистрация ноды в ComfyUI
+NODE_CLASS_MAPPINGS = {
+    "SimpleLatentPassthrough": SimpleLatentPassthrough
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "SimpleLatentPassthrough": "Simple Latent Passthrough (Dynamic)"
+}