Upload gradio_streamingllm.py
Browse files- gradio_streamingllm.py +3 -0
gradio_streamingllm.py
CHANGED
@@ -74,6 +74,9 @@ with gr.Blocks() as setting:
|
|
74 |
**cfg['setting_mirostat_eta'])
|
75 |
cfg['setting_mirostat_tau'] = gr.Number(label="Mirostat 目标熵", interactive=True,
|
76 |
**cfg['setting_mirostat_tau'])
|
|
|
|
|
|
|
77 |
|
78 |
# ========== 加载模型 ==========
|
79 |
cfg['model'] = StreamingLLM(model_path=cfg['setting_path'].value,
|
|
|
74 |
**cfg['setting_mirostat_eta'])
|
75 |
cfg['setting_mirostat_tau'] = gr.Number(label="Mirostat 目标熵", interactive=True,
|
76 |
**cfg['setting_mirostat_tau'])
|
77 |
+
with gr.Row(elem_classes='setting'):
|
78 |
+
cfg['setting_btn_vo_keep_last'] = gr.Number(label="旁白限制", interactive=True,
|
79 |
+
**cfg['setting_btn_vo_keep_last'])
|
80 |
|
81 |
# ========== 加载模型 ==========
|
82 |
cfg['model'] = StreamingLLM(model_path=cfg['setting_path'].value,
|