obliteratus / examples /preset_knowledge.yaml
pliny-the-prompter's picture
Upload 77 files
2bc8e46 verified
# Example: Find where knowledge is stored (FFNs + embeddings)
# Run with: obliteratus run examples/preset_knowledge.yaml
preset: knowledge
model:
name: gpt2
task: causal_lm
dtype: float32
device: cpu
dataset:
name: wikitext
subset: wikitext-2-raw-v1
split: test
text_column: text
output_dir: results/gpt2_knowledge