diff options
| author | Sj-Si <sjw.jetty@gmail.com> | 2024-01-11 21:37:35 +0000 |
|---|---|---|
| committer | Sj-Si <sjw.jetty@gmail.com> | 2024-01-11 21:37:35 +0000 |
| commit | 036500223de0a3caaa86360a8ad3ed301e4367b0 (patch) | |
| tree | f05f0d5fc503d9c35d57bad077a5dab1dfd6569e /modules/sd_hijack_utils.py | |
| parent | 0726a6e12e85a37d1e514f5603acf9f058c11783 (diff) | |
| parent | cb5b335acddd126d4f6c990982816c06beb0d6ae (diff) | |
| download | stable-diffusion-webui-gfx803-036500223de0a3caaa86360a8ad3ed301e4367b0.tar.gz stable-diffusion-webui-gfx803-036500223de0a3caaa86360a8ad3ed301e4367b0.tar.bz2 stable-diffusion-webui-gfx803-036500223de0a3caaa86360a8ad3ed301e4367b0.zip | |
Merge changes from dev
Diffstat (limited to 'modules/sd_hijack_utils.py')
| -rw-r--r-- | modules/sd_hijack_utils.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/modules/sd_hijack_utils.py b/modules/sd_hijack_utils.py index f8684475..79bf6e46 100644 --- a/modules/sd_hijack_utils.py +++ b/modules/sd_hijack_utils.py @@ -11,10 +11,14 @@ class CondFunc: break
except ImportError:
pass
- for attr_name in func_path[i:-1]:
- resolved_obj = getattr(resolved_obj, attr_name)
- orig_func = getattr(resolved_obj, func_path[-1])
- setattr(resolved_obj, func_path[-1], lambda *args, **kwargs: self(*args, **kwargs))
+ try:
+ for attr_name in func_path[i:-1]:
+ resolved_obj = getattr(resolved_obj, attr_name)
+ orig_func = getattr(resolved_obj, func_path[-1])
+ setattr(resolved_obj, func_path[-1], lambda *args, **kwargs: self(*args, **kwargs))
+ except AttributeError:
+ print(f"Warning: Failed to resolve {orig_func} for CondFunc hijack")
+ pass
self.__init__(orig_func, sub_func, cond_func)
return lambda *args, **kwargs: self(*args, **kwargs) def __init__(self, orig_func, sub_func, cond_func): |