From 39db2cf706d8e0e8f1a36c21a5d8312a614c41fa Mon Sep 17 00:00:00 2001
From: awsr <43862868+awsr@users.noreply.github.com>
Date: Sun, 10 May 2026 13:50:56 -0700
Subject: [PATCH] processing_class typing and defaultdict

Use defaultdict for network_data to match usage elsewhere
---
 modules/processing_class.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/modules/processing_class.py b/modules/processing_class.py
index c09fd0d62..bf6c268f0 100644
--- a/modules/processing_class.py
+++ b/modules/processing_class.py
@@ -1,8 +1,11 @@
+from __future__ import annotations
+
 import os
 import sys
 import inspect
 import hashlib
-from typing import Any
+from collections import defaultdict
+from typing import Any, TYPE_CHECKING
 from dataclasses import dataclass, field
 import numpy as np
 from PIL import Image, ImageOps
@@ -11,6 +14,9 @@ from modules.logger import log
 from modules.paths import resolve_output_path
 from modules.image.util import flatten
 
+if TYPE_CHECKING:
+    from modules.extra_networks import ExtraNetworkParams
+
 
 debug = log.trace if os.environ.get('SD_PROCESS_DEBUG', None) is not None else lambda *args, **kwargs: None
 
@@ -323,7 +329,9 @@ class StableDiffusionProcessing:
         self.negative_prompt_attention_masks = []
         self.disable_extra_networks = False
         self.iteration = 0
-        self.network_data = network_data or {}
+        self.network_data: defaultdict[str, list[ExtraNetworkParams]] = defaultdict(list)
+        if network_data is not None:
+            self.network_data |= network_data
 
         # initializers
         self.prompt = prompt