style

allenai · Oct 23, 2024 · bfd9b15 · bfd9b15
1 parent 0636d90
commit bfd9b15
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/rewardbench/models/__init__.py b/rewardbench/models/__init__.py
@@ -222,7 +222,6 @@
         "custom_dialogue": False,
         "model_type": "Seq. Classifier",
     },
-
 }
 
 DPO_MODEL_CONFIG = {

diff --git a/rewardbench/models/qrm.py b/rewardbench/models/qrm.py
@@ -5,10 +5,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint
-from transformers import (
-    LlamaModel,
-    LlamaPreTrainedModel,
-)
+from transformers import LlamaModel, LlamaPreTrainedModel
 from transformers.models.llama.modeling_llama import LLAMA_INPUTS_DOCSTRING
 from transformers.utils import ModelOutput, add_start_docstrings_to_model_forward
-Original file line number
+Diff line change
@@ Expand Up / @@ -222,7 +222,6 @@ @@
             "custom_dialogue": False,
             "model_type": "Seq. Classifier",
         },
     }
     DPO_MODEL_CONFIG = {
@@ Expand Down @@