fix-base-model-submission (#901)
Browse files- Base model only for Delta or Adapter (0d97e085e758bc706a0efb15bd9609800e33265b)
app.py
CHANGED
|
@@ -259,10 +259,10 @@ with main_block:
|
|
| 259 |
choices=[i.value.name for i in WeightType],
|
| 260 |
label="Weights type",
|
| 261 |
multiselect=False,
|
| 262 |
-
value=
|
| 263 |
interactive=True,
|
| 264 |
)
|
| 265 |
-
base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
|
| 266 |
|
| 267 |
with gr.Column():
|
| 268 |
with gr.Accordion(
|
|
@@ -316,6 +316,25 @@ with main_block:
|
|
| 316 |
outputs=chat_template_toggle,
|
| 317 |
)
|
| 318 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 319 |
submit_button.click(
|
| 320 |
add_new_eval,
|
| 321 |
[
|
|
|
|
| 259 |
choices=[i.value.name for i in WeightType],
|
| 260 |
label="Weights type",
|
| 261 |
multiselect=False,
|
| 262 |
+
value=WeightType.Original.value.name,
|
| 263 |
interactive=True,
|
| 264 |
)
|
| 265 |
+
base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)", interactive=False)
|
| 266 |
|
| 267 |
with gr.Column():
|
| 268 |
with gr.Accordion(
|
|
|
|
| 316 |
outputs=chat_template_toggle,
|
| 317 |
)
|
| 318 |
|
| 319 |
+
# The base_model_name_textbox interactivity and value reset function
|
| 320 |
+
def update_base_model_name_textbox(weight_type_value):
|
| 321 |
+
# Convert the dropdown value back to the corresponding WeightType Enum
|
| 322 |
+
weight_type_enum = WeightType[weight_type_value]
|
| 323 |
+
|
| 324 |
+
# Determine if the textbox should be interactive
|
| 325 |
+
interactive = weight_type_enum in [WeightType.Adapter, WeightType.Delta]
|
| 326 |
+
|
| 327 |
+
# Reset the value if weight type is "Original"
|
| 328 |
+
reset_value = "" if not interactive else None
|
| 329 |
+
|
| 330 |
+
return gr.update(interactive=interactive, value=reset_value)
|
| 331 |
+
|
| 332 |
+
weight_type.change(
|
| 333 |
+
fn=update_base_model_name_textbox,
|
| 334 |
+
inputs=[weight_type],
|
| 335 |
+
outputs=[base_model_name_textbox],
|
| 336 |
+
)
|
| 337 |
+
|
| 338 |
submit_button.click(
|
| 339 |
add_new_eval,
|
| 340 |
[
|