Skip to content

Commit 53e5b36

Browse files
authored
Revert "chore(deps): upgrade trl (#537)" (#539)
This reverts commit 6fc9f46. Signed-off-by: Dushyant Behl <dushyantbehl@in.ibm.com>
1 parent 6fc9f46 commit 53e5b36

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

docs/offline-data-preprocessing.md

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,6 @@ python scripts/offline_data_processing.py \
3737

3838
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.
3939

40-
**NOTE**: The offline data preprocessing script is not compatible with processing image datasets for vision models.
41-
4240
## Example Usage
4341
### Applying Chat Template
4442

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ dependencies = [
3434
"sentencepiece>=0.1.99,<0.3",
3535
"tokenizers>=0.13.3,<1.0",
3636
"tqdm>=4.66.2,<5.0",
37-
"trl>=0.13,<0.18",
37+
"trl>=0.13,<0.17",
3838
"peft>=0.8.0,<0.14",
3939
"protobuf>=5.28.0,<6.0.0",
4040
"datasets>=2.15.0,<4.0",

0 commit comments

Comments
 (0)