Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #14560 from Nuullll/condfunc-warning 

Handle CondFunc exception when resolving attributes
This commit is contained in:
AUTOMATIC1111 2024-01-07 08:23:08 +03:00 committed by GitHub
parent cab1d839b5 ec9acb3145
commit 30aa5e0a7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/sd_hijack_utils.py
View File

@ -11,10 +11,14 @@ class CondFunc:
break break
except ImportError: except ImportError:
pass pass
for attr_name in func_path[i:-1]: try:
resolved_obj = getattr(resolved_obj, attr_name) for attr_name in func_path[i:-1]:
orig_func = getattr(resolved_obj, func_path[-1]) resolved_obj = getattr(resolved_obj, attr_name)
setattr(resolved_obj, func_path[-1], lambda *args, **kwargs: self(*args, **kwargs)) 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) self.__init__(orig_func, sub_func, cond_func)
return lambda *args, **kwargs: self(*args, **kwargs) return lambda *args, **kwargs: self(*args, **kwargs)
def __init__(self, orig_func, sub_func, cond_func): def __init__(self, orig_func, sub_func, cond_func):