Update any_model.py

any_model.py

@@ -857,7 +857,7 @@ class AnyRewardModel(AnyModelForConditionalGeneration):
             end_index=end_index,  # size = (B,)
         )
         
-from transformers import AutoConfig, AutoModel
+from transformers import AutoConfig, AutoModel, AutoProcessor
     
 AutoConfig.register("any_model", AnyModelConfig)
 AutoModel.register(AnyModelConfig, AnyModelForConditionalGeneration)