Update hf_utils.py
Browse files- hf_utils.py +2 -2
hf_utils.py
CHANGED
|
@@ -125,8 +125,8 @@ class RewardModelConfig(PretrainedConfig):
|
|
| 125 |
class AutoModelForCausalLMWithRM(PreTrainedModel):
|
| 126 |
config_class = RewardModelConfig
|
| 127 |
|
| 128 |
-
def __init__(self, config: RewardModelConfig):
|
| 129 |
-
super().__init__(config)
|
| 130 |
self.config = config
|
| 131 |
pretrain_cfg = config.pretrain_cfg
|
| 132 |
pretrained = config.pretrained
|
|
|
|
| 125 |
class AutoModelForCausalLMWithRM(PreTrainedModel):
|
| 126 |
config_class = RewardModelConfig
|
| 127 |
|
| 128 |
+
def __init__(self, config: RewardModelConfig, *args, **kwargs):
|
| 129 |
+
super().__init__(config, *args, **kwargs)
|
| 130 |
self.config = config
|
| 131 |
pretrain_cfg = config.pretrain_cfg
|
| 132 |
pretrained = config.pretrained
|