Skip to content

Commit 68959ad

Browse files
committed
Squashed commit of the following:
commit 65eb45c Author: Quentin Gallouédec <[email protected]> Date: Mon Oct 6 13:07:18 2025 -0600 Apply style and revert change in `sft_video_llm` example (#4214) commit ae6837f Author: Sergio Paniego Blanco <[email protected]> Date: Mon Oct 6 18:40:18 2025 +0200 Removed tokenizer/processor creation from example scripts (#4211) commit 56a8f11 Author: Albert Villanova del Moral <[email protected]> Date: Mon Oct 6 17:45:44 2025 +0200 Replace setup with pyproject and fix packaging unintended modules (#4194) commit 5291015 Author: Sergio Paniego Blanco <[email protected]> Date: Mon Oct 6 16:04:06 2025 +0200 Remove `Optional` from `processing_class` in `PPOTrainer` (#4212) commit 0588b1f Author: Sergio Paniego Blanco <[email protected]> Date: Mon Oct 6 15:57:17 2025 +0200 Updated vLLM integration guide (#4162) Co-authored-by: Quentin Gallouédec <[email protected]> commit 45ee98b Author: Albert Villanova del Moral <[email protected]> Date: Mon Oct 6 11:14:54 2025 +0200 Replace unittest with pytest (#4188) commit 3800a6e Author: Albert Villanova del Moral <[email protected]> Date: Mon Oct 6 11:13:21 2025 +0200 Hotfix: Exclude transformers 4.57.0 for Python 3.9 (#4209) Co-authored-by: Sergio Paniego Blanco <[email protected]> commit 7ad9ce8 Author: Sergio Paniego Blanco <[email protected]> Date: Mon Oct 6 11:04:20 2025 +0200 Remove tokenizer creation from `sft` example script (#4197) commit 0c2dc14 Author: Albert Villanova del Moral <[email protected]> Date: Mon Oct 6 08:31:58 2025 +0200 Remove custome_container for building the docs (#4198) commit ced8b33 Author: burtenshaw <[email protected]> Date: Mon Oct 6 08:23:11 2025 +0200 [DOCS/FIX] lora without regrets - fix lr (#4207)
1 parent b07df79 commit 68959ad

File tree

6 files changed

+5
-5
lines changed

6 files changed

+5
-5
lines changed

examples/scripts/reward_modeling.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,6 @@
6767
get_kbit_device_map,
6868
get_peft_config,
6969
get_quantization_config,
70-
setup_chat_format,
7170
)
7271

7372

examples/scripts/sft_gpt_oss.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,6 @@ def main(script_args, training_args, model_args):
7575

7676
model = AutoModelForCausalLM.from_pretrained(model_args.model_name_or_path, **model_kwargs)
7777

78-
7978
# Load dataset
8079
dataset = load_dataset(script_args.dataset_name, name=script_args.dataset_config)
8180

tests/test_activation_offloading.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,6 +151,6 @@ def test_real_hf_model(self):
151151
grads2 = [p.grad.clone() for p in model.parameters()]
152152

153153
# Check outputs and gradients match
154-
self.assertTrue(torch.allclose(out1, out2, rtol=1e-5))
154+
assert torch.allclose(out1, out2, rtol=1e-5)
155155
for g1, g2 in zip(grads1, grads2, strict=True):
156-
self.assertTrue(torch.allclose(g1, g2, rtol=1e-5))
156+
assert torch.allclose(g1, g2, rtol=1e-5)

tests/test_cli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,11 @@
1414

1515

1616
import os
17+
import sys
1718
from io import StringIO
1819
from unittest.mock import patch
1920

21+
import pytest
2022
import yaml
2123

2224
from .testing_utils import TrlTestCase

tests/test_dpo_trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
# limitations under the License.
1414

1515
import re
16+
import sys
1617
from unittest.mock import MagicMock
1718

1819
import numpy as np

trl/scripts/dpo.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,6 @@
8080
get_peft_config,
8181
get_quantization_config,
8282
)
83-
from trl.trainer.utils import SIMPLE_CHAT_TEMPLATE
8483

8584

8685
logger = logging.get_logger(__name__)

0 commit comments

Comments
 (0)