PKU-Alignment
/

AnyRewardModel

Model card Files Files and versions Community

XuyaoWang commited on 5 days ago

Commit

018e2db

•

1 Parent(s): a152b8f

Update any_model.py

Files changed (1) hide show

any_model.py +2 -1

any_model.py CHANGED Viewed

@@ -861,4 +861,5 @@ from transformers import AutoConfig, AutoModel
 AutoConfig.register("any_model", AnyModelConfig)
 AutoModel.register(AnyModelConfig, AnyModelForConditionalGeneration)
-AutoModel.register(AnyModelConfig, AnyRewardModel)

 AutoConfig.register("any_model", AnyModelConfig)
 AutoModel.register(AnyModelConfig, AnyModelForConditionalGeneration)
+AutoModel.register(AnyModelConfig, AnyRewardModel)
+AutoProcessor.register("AnyModelProcessor", AnyModelProcessor)