Update src/auto_leaderboard/model_metadata_type.py

#199
src/auto_leaderboard/model_metadata_type.py CHANGED
@@ -206,10 +206,10 @@ TYPE_METADATA: Dict[str, ModelType] = {
206
  'dvruette/oasst-gpt-neox-20b-1000-steps': ModelType.FT,
207
  'dvruette/llama-13b-pretrained-dropout': ModelType.PT,
208
  'dvruette/llama-13b-pretrained': ModelType.PT,
209
- 'dvruette/llama-13b-pretrained-sft-epoch-1': ModelType.PT,
210
- 'dvruette/llama-13b-pretrained-sft-do2': ModelType.PT,
211
  'dvruette/oasst-gpt-neox-20b-3000-steps': ModelType.FT,
212
- 'dvruette/oasst-pythia-12b-pretrained-sft': ModelType.PT,
213
  'dvruette/oasst-pythia-6.9b-4000-steps': ModelType.FT,
214
  'dvruette/gpt-neox-20b-full-precision': ModelType.FT,
215
  'dvruette/oasst-llama-13b-1000-steps': ModelType.FT,
 
206
  'dvruette/oasst-gpt-neox-20b-1000-steps': ModelType.FT,
207
  'dvruette/llama-13b-pretrained-dropout': ModelType.PT,
208
  'dvruette/llama-13b-pretrained': ModelType.PT,
209
+ 'dvruette/llama-13b-pretrained-sft-epoch-1': ModelType.FT,
210
+ 'dvruette/llama-13b-pretrained-sft-do2': ModelType.FT,
211
  'dvruette/oasst-gpt-neox-20b-3000-steps': ModelType.FT,
212
+ 'dvruette/oasst-pythia-12b-pretrained-sft': ModelType.FT,
213
  'dvruette/oasst-pythia-6.9b-4000-steps': ModelType.FT,
214
  'dvruette/gpt-neox-20b-full-precision': ModelType.FT,
215
  'dvruette/oasst-llama-13b-1000-steps': ModelType.FT,