Skip to content

Commit

Permalink
restore mpt eval
Browse files Browse the repository at this point in the history
  • Loading branch information
bmosaicml committed Nov 13, 2023
1 parent a1f8ffb commit 6014dd4
Showing 1 changed file with 11 additions and 4 deletions.
15 changes: 11 additions & 4 deletions scripts/eval/yamls/mpt_eval.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ models:
name: ${tokenizer_name}
kwargs:
model_max_length: ${max_seq_len}
load_path: # Add your (optional) Composer checkpoint path here!
model:
name: mpt_causal_lm
init_device: mixed
Expand All @@ -25,7 +24,9 @@ models:
attn_config:
attn_impl: triton

device_eval_batch_size: 4
load_path: # Add your non-optional Composer checkpoint path here! (must not be empty)

device_eval_batch_size: 16

# FSDP config for model sharding
fsdp_config:
Expand All @@ -34,5 +35,11 @@ fsdp_config:
forward_prefetch: True
limit_all_gathers: True

icl_tasks: 'eval/yamls/tasks_v0.1.yaml'
eval_gauntlet: 'eval/yamls/eval_gauntlet_v0.1.yaml'
icl_tasks:
-
label: jeopardy
dataset_uri: eval/local_data/world_knowledge/jeopardy_all.jsonl # ADD YOUR OWN DATASET URI
num_fewshot: [0]
icl_task_type: language_modeling
continuation_delimiter: "\nAnswer: " # this separates questions from answers
has_categories: true

0 comments on commit 6014dd4

Please sign in to comment.