pip install -q datasets trl peft bitsandbytes sentencepiece wandb