frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove Optional from processing_class in PPOTrainer (#4212)

Browse Source
This commit is contained in:
Sergio Paniego Blanco
2025-10-06 16:04:06 +02:00
committed by GitHub
parent 0588b1f01d
commit 529101537f
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
trl/trainer/ppo_trainer.py
View File

@ -146,9 +146,7 @@ class PPOTrainer(BaseTrainer):
def __init__(
self,
args: PPOConfig,
processing_class: Optional[
Union[PreTrainedTokenizerBase, BaseImageProcessor, FeatureExtractionMixin, ProcessorMixin]
],
processing_class: Union[PreTrainedTokenizerBase, BaseImageProcessor, FeatureExtractionMixin, ProcessorMixin],
model: nn.Module,
ref_model: Optional[nn.Module],
reward_model: nn.Module,