Rename modeling_custom_old.py to modeling_custom.py

modeling_custom_old.py → modeling_custom.py

@@ -173,7 +173,7 @@ class LlamaForRewardModelWithGating(LlamaForSequenceClassification):
                 torch.transpose(reward_quantiles_all_adjusted, 1, 2)
             ).sum(1)
 
-            rewards_expectation = rewards.float().mean(dim=2)
+        rewards_expectation = rewards.float().mean(dim=2)
 
         rewards_expectation_adjusted = rewards_expectation @ self.reward_transform_matrix
         score = torch.sum(gating_output * rewards_expectation_adjusted, dim=1, keepdim=True)