stellaathena commited on
Commit
f28c1e8
1 Parent(s): 0f4a719

Update src/auto_leaderboard/model_metadata_type.py

Browse files
src/auto_leaderboard/model_metadata_type.py CHANGED
@@ -201,7 +201,7 @@ TYPE_METADATA: Dict[str, ModelType] = {
201
  "dvruette/llama-13b-pretrained-sft-epoch-1": ModelType.PT,
202
  "dvruette/llama-13b-pretrained-sft-do2": ModelType.PT,
203
  "dvruette/oasst-gpt-neox-20b-3000-steps": ModelType.SFT,
204
- "dvruette/oasst-pythia-12b-pretrained-sft": ModelType.PT,
205
  "dvruette/oasst-pythia-6.9b-4000-steps": ModelType.SFT,
206
  "dvruette/gpt-neox-20b-full-precision": ModelType.SFT,
207
  "dvruette/oasst-llama-13b-1000-steps": ModelType.SFT,
 
201
  "dvruette/llama-13b-pretrained-sft-epoch-1": ModelType.PT,
202
  "dvruette/llama-13b-pretrained-sft-do2": ModelType.PT,
203
  "dvruette/oasst-gpt-neox-20b-3000-steps": ModelType.SFT,
204
+ "dvruette/oasst-pythia-12b-pretrained-sft": ModelType.SFT,
205
  "dvruette/oasst-pythia-6.9b-4000-steps": ModelType.SFT,
206
  "dvruette/gpt-neox-20b-full-precision": ModelType.SFT,
207
  "dvruette/oasst-llama-13b-1000-steps": ModelType.SFT,