jdchang commited on
Commit
b5a4dc3
·
verified ·
1 Parent(s): 6aebb2a

Update hf_utils.py

Browse files
Files changed (1) hide show
  1. 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