Merge pull request #26 from Control-derek/derek

add self.
lucidrains · Mar 27, 2024 · 72291ff · 72291ff
2 parents 0649dfb + 72c2dca
commit 72291ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/self_rewarding_lm_pytorch/self_rewarding_lm_pytorch.py b/self_rewarding_lm_pytorch/self_rewarding_lm_pytorch.py
@@ -506,7 +506,7 @@ def generate_reward(
         reward_prompt = repeat(reward_prompt, 'n -> b n', b = self.num_evals_to_average)
 
         reward_prompt = reward_prompt.to(device)
-        self_reward_model = self_reward_model.to(device)
+        self_reward_model = self.self_reward_model.to(device)
 
         reward_responses = sample(
             self_reward_model,