Hugging Face's logo

PKU-Alignment
/
AnyRewardModel

Any-to-Any
PyTorch
English
any_model
custom_code
Model card Files
xet
Community
6
XuyaoWang commited on
Commit
018e2db
·
verified ·
1 Parent(s): a152b8f

Update any_model.py

Browse files
Files changed (1) hide show
  1. any_model.py +2 -1
any_model.py CHANGED
@@ -861,4 +861,5 @@ from transformers import AutoConfig, AutoModel
861
 
862
  AutoConfig.register("any_model", AnyModelConfig)
863
  AutoModel.register(AnyModelConfig, AnyModelForConditionalGeneration)
864
- AutoModel.register(AnyModelConfig, AnyRewardModel)
 
 
861
 
862
  AutoConfig.register("any_model", AnyModelConfig)
863
  AutoModel.register(AnyModelConfig, AnyModelForConditionalGeneration)
864
+ AutoModel.register(AnyModelConfig, AnyRewardModel)
865
+ AutoProcessor.register("AnyModelProcessor", AnyModelProcessor)