From ba5464d3684615a0d7045e80267d409c633b6ea1 Mon Sep 17 00:00:00 2001 From: Dushyant Behl Date: Fri, 25 Apr 2025 10:16:06 +0530 Subject: [PATCH] Revert "chore(deps): upgrade trl (#537)" This reverts commit 6fc9f46e0116143de2b87b1f23de0ab941367a04. Signed-off-by: Dushyant Behl --- docs/offline-data-preprocessing.md | 2 -- pyproject.toml | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/docs/offline-data-preprocessing.md b/docs/offline-data-preprocessing.md index 53e77d195..f235d5aa4 100644 --- a/docs/offline-data-preprocessing.md +++ b/docs/offline-data-preprocessing.md @@ -37,8 +37,6 @@ python scripts/offline_data_processing.py \ Additionally, once the offline data processing is complete, users can leverage the shards stored in `output_dir` for tuning by passing it through the `--training_data_path` flag or passing it via `data_paths` argument in data config yaml, provided they find the sharded datasets beneficial for training. -**NOTE**: The offline data preprocessing script is not compatible with processing image datasets for vision models. - ## Example Usage ### Applying Chat Template diff --git a/pyproject.toml b/pyproject.toml index 09732281c..f67483d55 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,7 +34,7 @@ dependencies = [ "sentencepiece>=0.1.99,<0.3", "tokenizers>=0.13.3,<1.0", "tqdm>=4.66.2,<5.0", -"trl>=0.13,<0.18", +"trl>=0.13,<0.17", "peft>=0.8.0,<0.14", "protobuf>=5.28.0,<6.0.0", "datasets>=2.15.0,<4.0",