torch transformers datasets peft trl bitsandbytes accelerate scikit-learn