| class Bonso_Switch_Model: |
| """Cambia entre dos modelos (MODEL) segun la casilla `switch`. |
| |
| Pensado para encajar con `Bonso Booleano`: su booleano sale por el |
| socket y entra aqui por `switch`. |
| |
| switch = True -> devuelve true_model |
| switch = False -> devuelve false_model |
| |
| Tres cosas que hay que dejar claras para no romperlo: |
| |
| - **El modelo pasa tal cual**: se devuelve la misma referencia que ha |
| llegado. Ni clone, ni copia, ni transformacion, ni operacion alguna. |
| Si la entrada es un `ModelPatcher` de ComfyUI, la salida es ese mismo |
| objeto. |
| - **La salida es `MODEL`, no `*`**: asi el servidor valida de verdad el |
| enlace (`Return type mismatch between linked nodes` si intentas |
| enchufar esto a una entrada CLIP/CONDITIONING) y el frontend mantiene |
| la linea morada. Un ANY disfrazado no avisaria de nada. |
| - **Las dos entradas de modelo son `optional`**: ComfyUI valida el |
| prompt COMPLETO antes de ejecutar, asi que una rama rota (nodo en |
| bypass, entrada que falta, checkpoint que no existe) reventaria el |
| trabajo aunque no tocara. Por eso `js/bonso_switch_model.js` **saca la |
| rama inactiva del prompt** antes de encolar: lo que no se usa no se |
| valida ni se ejecuta. Sin ese JS (o si el valor de `switch` no se |
| puede saber) se envian las dos ramas, que es el comportamiento de |
| siempre. |
| |
| Si falta la rama que SI toca, el nodo dice cual es con un error claro |
| en vez de devolver `None` y estrellar el KSampler mas abajo. |
| """ |
|
|
| @classmethod |
| def INPUT_TYPES(cls): |
| return { |
| "required": { |
| "switch": ( |
| "BOOLEAN", |
| { |
| "default": True, |
| "tooltip": "True -> true_model, False -> false_model.", |
| }, |
| ), |
| }, |
| "optional": { |
| "true_model": ( |
| "MODEL", |
| {"tooltip": "Modelo que sale con switch en True."}, |
| ), |
| "false_model": ( |
| "MODEL", |
| {"tooltip": "Modelo que sale con switch en False."}, |
| ), |
| }, |
| } |
|
|
| RETURN_TYPES = ("MODEL",) |
| RETURN_NAMES = ("model",) |
| FUNCTION = "switch_model" |
| CATEGORY = "Bonso Tools" |
| DESCRIPTION = ( |
| "Devuelve true_model si switch es True y false_model si es False. " |
| "El modelo sale igual que ha entrado: misma referencia, sin clonar. " |
| "La rama que no toca queda fuera del prompt y no se valida ni se ejecuta." |
| ) |
|
|
| def switch_model(self, switch=True, true_model=None, false_model=None): |
| if switch: |
| if true_model is None: |
| raise ValueError( |
| "[Bonso Switch Model] switch=True pero 'true_model' no esta " |
| "conectado: conecta el modelo de la rama activa o cambia switch a False." |
| ) |
| return (true_model,) |
| if false_model is None: |
| raise ValueError( |
| "[Bonso Switch Model] switch=False pero 'false_model' no esta " |
| "conectado: conecta el modelo de la rama activa o cambia switch a True." |
| ) |
| return (false_model,) |
|
|
|
|
| NODE_CLASS_MAPPINGS = { |
| "Bonso_Switch_Model": Bonso_Switch_Model, |
| } |
| NODE_DISPLAY_NAME_MAPPINGS = { |
| "Bonso_Switch_Model": "Bonso Switch Model", |
| } |
|
|
