Update requirements and setup scripts for consistency and version con…

[KartikB4B]

Update requirements and setup scripts for consistency and version control.

Requirements file -->

accelerate==1.0.1
datasets==3.2.0
safetensors==0.5.2
transformers==4.48.2
trl==0.9.6
hf_transfer==0.1.9

SetUp Script -->

#!/bin/bash

Install PyTorch

pip install torch==2.6.0

Install wheel, packaging, and ninja

pip install packaging==24.2 ninja==1.11.1.3

Install flash-attn and deepspeed

pip install flash-attn==2.7.4.post1 --no-build-isolation
pip install deepspeed==0.16.3

Install requirements from requirements.txt

pip install -r requirements.txt