diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-09-09 08:38:42 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-09-09 08:38:49 +0300 |
| commit | 259768f27fc4da61000610bc81a16f0152b36550 (patch) | |
| tree | 373b3a70e24e8b6cbc694d23372be6d3e519682b /modules/scripts.py | |
| parent | 741e8ecb7d2192ed3bdeabab1b8827d25fafa543 (diff) | |
fix the bug in script-info API
Diffstat (limited to 'modules/scripts.py')
| -rw-r--r-- | modules/scripts.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/modules/scripts.py b/modules/scripts.py index e8518ad0..f1f17a5f 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -491,11 +491,15 @@ class ScriptRunner: arg_info = api_models.ScriptArg(label=control.label or "")
- for field in ("value", "minimum", "maximum", "step", "choices"):
+ for field in ("value", "minimum", "maximum", "step"):
v = getattr(control, field, None)
if v is not None:
setattr(arg_info, field, v)
+ choices = getattr(control, 'choices', None) # as of gradio 3.41, some items in choices are strings, and some are tuples where the first elem is the string
+ if choices is not None:
+ setattr(arg_info, 'choices', [x[0] if isinstance(x, tuple) else x for x in choices])
+
api_args.append(arg_info)
script.api_info = api_models.ScriptInfo(
|