Fix: add kwargs in reward classifier __init__()

This commit is contained in:
Khalil Meftah
2026-04-14 11:13:43 +02:00
parent a8838c081b
commit e6d282108d
@@ -107,6 +107,7 @@ class Classifier(PreTrainedPolicy):
def __init__(
self,
config: RewardClassifierConfig,
**kwargs,
):
from transformers import AutoModel