Skip to content

Commit

Permalink
add model && fix code quality error
Browse files Browse the repository at this point in the history
  • Loading branch information
Mghao committed Dec 11, 2024
1 parent e8d6439 commit b44a460
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rewardbench/models/inform.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -84,4 +85,3 @@ def forward(
hidden_states=transformer_outputs.hidden_states,
attentions=transformer_outputs.attentions,
)

0 comments on commit b44a460

Please sign in to comment.