diff --git a/requirements.txt b/requirements.txt index f8969caa..ce3994a8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ datasets>=2.15.0 numba numpy==1.26.4 rich -instructlab-dolomite @ git+https://github.com/instructlab/GPTDolomite.git@main +instructlab-dolomite trl>=0.9.4 peft pydantic>=2.7.0