mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
processing_class typing and defaultdict
Use defaultdict for network_data to match usage elsewhere
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user