fix #2164
This commit is contained in:
1 parent
7bf6612f4a
commit
4b2d11ec28
1 file changed
+1
@@ -36,6 +36,7 @@ class CustomDPOTrainer(DPOTrainer):
|
||||
self.precompute_ref_log_probs = False
|
||||
self._precomputed_train_ref_log_probs = False
|
||||
self._precomputed_eval_ref_log_probs = False
|
||||
self._peft_has_been_casted_to_bf16 = False
|
||||
|
||||
self.ref_model = ref_model
|
||||
self.beta = beta
|
||||
|
||||
Reference in new issue
Block a user