cxllm-vendors/stable-diffusion-webui-forge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update gradio_compile.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 22:31:06 -08:00
parent e76ab46ec9
commit f0603b3baa
1 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules_forge/gradio_compile.py
+20 -10
View File
@@ -1,14 +1,24 @@
def gradio_compile(items, prefix): def gradio_compile(items, prefix):
for k, v in items["required"].items(): for k, v in items["required"].items():
if len(v) == 2: t = v[0]
t, d = v d = v[1] if len(v) > 1 else None
if t == 'INT': name = (prefix + '_' + k).replace(' ', '_').lower()
name = (prefix + '_' + k).replace(' ', '_').lower() title = name.replace('_', ' ').capitalize()
title = name.replace('_', ' ').capitalize()
default = int(d['default']) if t == 'INT':
min = int(d['min']) default = int(d['default'])
max = int(d['max']) min = int(d['min'])
step = int(d.get('step', 1)) max = int(d['max'])
print(f'{name} = gr.Slider(label=\'{title}\', minimum={min}, maximum={max}, step={step}, value={default})') step = int(d.get('step', 1))
print(f'{name} = gr.Slider(label=\'{title}\', minimum={min}, maximum={max}, step={step}, value={default})')
elif t == 'FLOAT':
default = float(d['default'])
min = float(d['min'])
max = float(d['max'])
step = float(d.get('step', 0.001))
print(f'{name} = gr.Slider(label=\'{title}\', minimum={min}, maximum={max}, step={step}, value={default})')
else:
print('error ' + t)
return return