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 |
|