Stella Biderman cbdb772eb9 Add multi-GPU support with --gpus flag ...

Adds --gpus flag to obliterate, run, and tourney commands for controlling
which GPUs to use (sets CUDA_VISIBLE_DEVICES). Works both locally and with
--remote. Models are automatically split across selected GPUs via
accelerate's device_map="auto". Also adds gpus field to remote YAML config.

2026-03-12 12:35:29 -04:00

..

full_study.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

gpt2_head_ablation.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

gpt2_layer_ablation.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

preset_attention.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

preset_knowledge.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

preset_quick.yaml

Add files via upload

2026-03-04 12:38:18 -08:00

remote_gpu_node.yaml

Add multi-GPU support with --gpus flag

2026-03-12 12:35:29 -04:00