not being cast properly every time, swap to ints

This commit is contained in:
papuSpartan 2023-05-03 21:10:31 -05:00
parent e960781511
commit f0efc8c211

View File

@ -488,7 +488,7 @@ options_templates.update(options_section(('token_merging', 'Token Merging'), {
False, "Merge mlp",
gr.Checkbox
),
"token_merging_maximum_down_sampling": OptionInfo(1, "Maximum down sampling", gr.Radio, lambda: {"choices": ['1', '2', '4', '8']}),
"token_merging_maximum_down_sampling": OptionInfo(1, "Maximum down sampling", gr.Radio, lambda: {"choices": [1, 2, 4, 8]}),
"token_merging_stride_x": OptionInfo(
2, "Stride - X",
gr.Slider, {"minimum": 2, "maximum": 8, "step": 2}