model:
  name: gpt2
  task: causal_lm
  dtype: float16
  device: cuda

dataset:
  name: Salesforce/wikitext
  subset: wikitext-2-raw-v1
  split: test
  text_column: text
  max_samples: 25

strategies:
  - name: layer_removal
    params: {}

metrics:
  - perplexity

batch_size: 4
max_length: 256
output_dir: results/gpt2_gpu_quick
