Spaces:
Running
Running
Use KL divergence by default
Browse files
app.py
CHANGED
|
@@ -76,7 +76,7 @@ if not compact_layout:
|
|
| 76 |
)
|
| 77 |
|
| 78 |
model_name = st.selectbox("Model", ["distilgpt2", "gpt2", "EleutherAI/gpt-neo-125m"])
|
| 79 |
-
metric_name = st.selectbox("Metric", ["KL divergence", "NLL loss"], index=
|
| 80 |
|
| 81 |
tokenizer = st.cache_resource(AutoTokenizer.from_pretrained, show_spinner=False)(model_name, use_fast=False)
|
| 82 |
|
|
|
|
| 76 |
)
|
| 77 |
|
| 78 |
model_name = st.selectbox("Model", ["distilgpt2", "gpt2", "EleutherAI/gpt-neo-125m"])
|
| 79 |
+
metric_name = st.selectbox("Metric", ["KL divergence", "NLL loss"], index=0)
|
| 80 |
|
| 81 |
tokenizer = st.cache_resource(AutoTokenizer.from_pretrained, show_spinner=False)(model_name, use_fast=False)
|
| 82 |
|