File size: 2,889 Bytes
cb3354e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import gradio as gr
import modelscope_studio.components.antd as antd
import modelscope_studio.components.base as ms

from config import *


def SettingsMenu():
    state = gr.State({"open": True})

    # Use a Card or just a Flex container for the sidebar
    with antd.Card(title="Settings", size="small", bordered=True):
        with antd.Form(value=DEFAULT_SETTINGS) as settings_form:
            with antd.Form.Item(form_name="model", label="Model"):
                with antd.Select(options=MODEL_OPTIONS):
                    with ms.Slot(
                        "labelRender",
                        params_mapping="""(option) => ({
                                label: option.label,
                                link: { href: window.MODEL_OPTIONS_MAP[option.value].link },
                            })""",
                    ):
                        antd.Typography.Text(as_item="label")

            with antd.Form.Item(
                form_name="temperature",
                label="Temperature",
                elem_classes="setting-form-temperature",
            ):
                antd.Slider(
                    elem_style=dict(flex=1, marginRight=14),
                    min=MIN_TEMPERATURE,
                    max=MAX_TEMPERATURE,
                    step=0.05,
                    tooltip=dict(formatter="(v) => `${v}`"),
                )
                antd.InputNumber(
                    max=MAX_TEMPERATURE,
                    min=MIN_TEMPERATURE,
                    step=0.05,
                    elem_style=dict(width=65),
                )

            with antd.Form.Item(
                form_name="max_new_tokens",
                label="Max new tokens",
                elem_classes="setting-form-max-new-tokens",
            ):
                antd.Slider(
                    elem_style=dict(flex=1, marginRight=14),
                    min=MIN_MAX_NEW_TOKENS,
                    max=MAX_MAX_NEW_TOKENS,
                    step=64,
                    tooltip=dict(formatter="(v) => `${v}`"),
                )
                antd.InputNumber(
                    max=MAX_MAX_NEW_TOKENS,
                    min=MIN_MAX_NEW_TOKENS,
                    step=64,
                    elem_style=dict(width=85),
                )

            with antd.Form.Item(
                form_name="top_p",
                label="Top_p",
                elem_classes="setting-form-top-p",
            ):
                antd.Slider(
                    elem_style=dict(flex=1, marginRight=14),
                    min=MIN_TOP_P,
                    max=MAX_TOP_P,
                    step=0.05,
                    tooltip=dict(formatter="(v) => `${v}`"),
                )
                antd.InputNumber(
                    max=MAX_TOP_P, min=MIN_TOP_P, step=0.05, elem_style=dict(width=65)
                )

    return state, settings_form