diff --git a/rewardbench/models/inform.py b/rewardbench/models/inform.py index a2d76d2..fe17266 100644 --- a/rewardbench/models/inform.py +++ b/rewardbench/models/inform.py @@ -18,6 +18,7 @@ from transformers import LlamaPreTrainedModel, LlamaModel from transformers.modeling_outputs import SequenceClassifierOutputWithPast + class INFORMForSequenceClassification(LlamaPreTrainedModel): def __init__(self, config): super().__init__(config) @@ -84,4 +85,3 @@ def forward( hidden_states=transformer_outputs.hidden_states, attentions=transformer_outputs.attentions, ) - \ No newline at end of file