add model inform

rewardbench/models/inform.py

@@ -12,11 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import List, Optional, Union
-
+from typing import List, Optional
 import torch
 import torch.nn as nn
-from transformers import LlamaPreTrainedModel,LlamaConfig, LlamaModel
+from transformers import LlamaPreTrainedModel, LlamaModel
 from transformers.modeling_outputs import SequenceClassifierOutputWithPast
 
 class INFORMForSequenceClassification(LlamaPreTrainedModel):
@@ -33,7 +32,6 @@ def __init__(self, config):
         # Initialize weights and apply final processing
         self.post_init()
 
-
     def forward(
         self,
         input_ids: Optional[torch.LongTensor] = None,
@@ -85,4 +83,5 @@ def forward(
             past_key_values=transformer_outputs.past_key_values,
             hidden_states=transformer_outputs.hidden_states,
             attentions=transformer_outputs.attentions,
-        )
+        )
+