fix file name

allenai · Feb 7, 2025 · 15addc9 · 15addc9
1 parent 1221ccd
commit 15addc9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rewardbench/models/__init__.py b/rewardbench/models/__init__.py
@@ -29,7 +29,7 @@
 from .grm import GRewardModel, GRMPipeline
 from .inform import INFORMForSequenceClassification
 from .internlm import InternLMPipeline
-from .lenovo import LDLRewardModel27B, LenovoPipeline
+from .ldlreward import LDLPipeline, LDLRewardModel27B
 from .openassistant import *  # noqa
 from .openbmb import LlamaRewardModel, OpenBMBPipeline
 from .pairrm import DebertaV2PairRM, PairRMPipeline
@@ -53,9 +53,9 @@
         "custom_dialogue": False,
         "model_type": "Seq. Classifier",
     },
-    "lenovo/LDL-Reward-Gemma-2-27B-v0.1": {
+    "ShikaiChen/LDL-Reward-Gemma-2-27B-v0.1": {
         "model_builder": LDLRewardModel27B.from_pretrained,
-        "pipeline_builder": LenovoPipeline,
+        "pipeline_builder": LDLPipeline,
         "quantized": False,
         "custom_dialogue": False,
         "model_type": "Seq. Classifier",

diff --git a/rewardbench/models/lenovo.py → rewardbench/models/ldlreward.py b/rewardbench/models/lenovo.py → rewardbench/models/ldlreward.py
@@ -192,7 +192,7 @@ def from_pretrained(cls, load_directory, device_map=None, *model_args, **kwargs)
         return model
 
 
-class LenovoPipeline:
+class LDLPipeline:
     def __init__(self, task, model, tokenizer):
         self.task = task
         self.model = model.eval()