reasoning-gym/examples/trl/requirements.txt

8 lines
98 B
Text

torch>=2.6.0
datasets
peft
transformers
trl
wandb
huggingface_hub
flash-attn --no-build-isolation