Fix edit node: None-safe steps/seed, wire steps/seed refs to Edit Image

app.py

@@ -38,7 +38,7 @@ _loaded_steps = None
 def _ensure_adapter(steps: int):
     """Load exactly one Pruna LoRA adapter matching the requested step count."""
     global _loaded_steps
-    steps = 5 if int(steps) == 5 else 8
+    steps = 5 if steps is not None and int(steps) == 5 else 8
     if _loaded_steps != steps:
         pipe.unload_lora_weights()
         pipe.load_lora_weights(
@@ -49,6 +49,13 @@ def _ensure_adapter(steps: int):
     return steps
 
 
+def _seed(seed) -> int:
+    try:
+        return int(seed)
+    except (TypeError, ValueError):
+        return 42
+
+
 def _coerce_image(image):
     """Accept a PIL image, a filepath, or a Gradio/Workflow file-ref dict."""
     if image is None or image == "":
@@ -79,7 +86,7 @@ def generate(prompt: str, steps: int = 8, seed: int = 42) -> dict:
         prompt=prompt,
         width=1024,
         height=1024,
-        generator=torch.Generator("cuda").manual_seed(int(seed)),
+        generator=torch.Generator("cuda").manual_seed(_seed(seed)),
         num_inference_steps=steps,
         sigmas=SIGMAS[steps],
         true_cfg_scale=1.0,
@@ -95,7 +102,7 @@ def edit(image, prompt: str, steps: int = 8, seed: int = 42) -> dict:
     out = pipe(
         prompt=prompt,
         image=_coerce_image(image),
-        generator=torch.Generator("cuda").manual_seed(int(seed)),
+        generator=torch.Generator("cuda").manual_seed(_seed(seed)),
         num_inference_steps=steps,
         sigmas=SIGMAS[steps],
         true_cfg_scale=1.0,

workflow.json

@@ -119,6 +119,8 @@
     {"id": "e4", "from_node_id": "ref_image", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "image", "type": "image"},
     {"id": "e7", "from_node_id": "ref_edit_prompt", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "prompt", "type": "text"},
     {"id": "e5", "from_node_id": "op_generate", "from_port_id": "out_0", "to_node_id": "sub_generated", "to_port_id": "in", "type": "image"},
+    {"id": "e8", "from_node_id": "ref_steps", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "steps", "type": "number"},
+    {"id": "e9", "from_node_id": "ref_seed", "from_port_id": "out", "to_node_id": "op_edit", "to_port_id": "seed", "type": "number"},
     {"id": "e6", "from_node_id": "op_edit", "from_port_id": "out_0", "to_node_id": "sub_edited", "to_port_id": "in", "type": "image"}
   ]
 }