notquitenothing
/
auto_sd_docker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Include compatible_samplers method and client output

This commit is contained in:
DepFA 2022-09-27 18:46:42 +01:00 committed by GitHub
parent 0f703af695
commit 584df9485d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
modules/scripts.py
View File

@ -56,7 +56,12 @@ class Script:
def describe(self): def describe(self):
return "" return ""
# If your script is only compatible with a subset of samplers, return the sampler
# names here, an empty list indicates compatibility with all samplers, for a full
# listing of current sampler names import modules.sd_samplers.samplers
def compatible_samplers(self):
return []
scripts_data = [] scripts_data = []
@ -133,6 +138,10 @@ class ScriptRunner:
control.custom_script_source = os.path.basename(script.filename) control.custom_script_source = os.path.basename(script.filename)
control.visible = False control.visible = False
concat_compatible_samplers = ",".join(script.compatible_samplers())
inputs += [gr.HTML(value='<span class="customScriptAttrs" data-supported-samplers="{}"></span>'.format(concat_compatible_samplers))]
inputs[-1].visible=False
inputs += controls inputs += controls
script.args_to = len(inputs) script.args_to = len(inputs)
@ -180,7 +189,7 @@ class ScriptRunner:
if script is None: if script is None:
return None return None
script_args = args[script.args_from:script.args_to] script_args = args[script.args_from+1:script.args_to]
processed = script.run(p, *script_args) processed = script.run(p, *script_args)
shared.total_tqdm.clear() shared.total_tqdm.clear()