Added missing token kwarg in Peft model loading (#1825)

huggingface · Jul 10, 2024 · 9e9dc96 · 9e9dc96
1 parent 7ddef5c
commit 9e9dc96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/models/modeling_base.py b/trl/models/modeling_base.py
@@ -215,7 +215,7 @@ class and the arguments that are specific to trl models. The kwargs
 
                 # Wrap the pretrained model with the trained peft adapter
                 pretrained_model = PeftModel.from_pretrained(
-                    pretrained_model, pretrained_model_name_or_path, is_trainable=is_trainable
+                    pretrained_model, pretrained_model_name_or_path, is_trainable=is_trainable, token=token
                 )
                 logging.info("Trained peft adapter loaded")
             else: