cifkao commited on
Commit
6f46ddf
·
1 Parent(s): f90dfb4

Use KL divergence by default

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=1)
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