Skip to content

Commit

Permalink
fix processing with nested calls
Browse files Browse the repository at this point in the history
Signed-off-by: Vladimir Mandic <[email protected]>
  • Loading branch information
vladmandic committed Dec 31, 2024
1 parent b37dffd commit 5f69cd0
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 8 deletions.
3 changes: 2 additions & 1 deletion CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,8 @@ NYE refresh release with quite a few optimizatios and bug fixes...
- enable debug logging by default
- image width/height calculation when doing img2img
- corrections with batch processing
- hires with refiner prompt and batch processing
- hires with refiner prompt and batch processing
- processing with nested calls

## Update for 2024-12-24

Expand Down
3 changes: 2 additions & 1 deletion installer.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ class Dot(dict): # dot notation access to dictionary attributes
diffusers_commit = "unknown"
extensions_commit = {
'sd-webui-controlnet': 'ecd33eb',
'adetailer': 'a89c01d'
# 'adetailer': 'a89c01d'
# 'stable-diffusion-webui-images-browser': '27fe4a7',
}

Expand Down Expand Up @@ -1075,6 +1075,7 @@ def set_environment():
os.environ.setdefault('TF_CPP_MIN_LOG_LEVEL', '2')
os.environ.setdefault('TF_ENABLE_ONEDNN_OPTS', '0')
os.environ.setdefault('USE_TORCH', '1')
os.environ.setdefault('TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD', '1')
os.environ.setdefault('UVICORN_TIMEOUT_KEEP_ALIVE', '60')
os.environ.setdefault('KINETO_LOG_LEVEL', '3')
os.environ.setdefault('DO_NOT_TRACK', '1')
Expand Down
16 changes: 10 additions & 6 deletions modules/processing_class.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,18 +116,23 @@ def __init__(self,
# overrides
override_settings: Dict[str, Any] = {},
override_settings_restore_afterwards: bool = True,
task_args: Dict[str, Any] = {},
ops: List[str] = [],
# metadata
extra_generation_params: Dict[Any, Any] = {},
# extra_generation_params: Dict[Any, Any] = {},
# task_args: Dict[str, Any] = {},
# ops: List[str] = [],
**kwargs,
):

for k, v in kwargs.items():
setattr(self, k, v)

# extra args set by processing loop
self.task_args = task_args
self.task_args = {}
self.extra_generation_params = {}

# state items
self.state: str = ''
self.ops = ops
self.ops = []
self.skip = []
self.color_corrections = []
self.is_control = False
Expand Down Expand Up @@ -203,7 +208,6 @@ def __init__(self,
self.do_not_save_samples = do_not_save_samples
self.do_not_save_grid = do_not_save_grid
self.override_settings_restore_afterwards = override_settings_restore_afterwards
self.extra_generation_params = extra_generation_params
self.eta = eta
self.cfg_scale = cfg_scale
self.cfg_end = cfg_end
Expand Down

0 comments on commit 5f69cd0

Please sign in to comment.