Update modeling_base.py (#2419)

huggingface · Nov 30, 2024 · 148b592 · 148b592
1 parent d6a8f2c
commit 148b592
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trl/models/modeling_base.py b/trl/models/modeling_base.py
@@ -619,7 +619,7 @@ def create_reference_model(
     """
     if is_deepspeed_zero3_enabled():
         raise ValueError(
-            "DeepSpeed ZeRO-3 is enabled and is not compatible with `create_reference_model()`. Please instantiate your reference model directly with `AutoCausalLM.from_pretrained()`."
+            "DeepSpeed ZeRO-3 is enabled and is not compatible with `create_reference_model()`. Please instantiate your reference model directly with `AutoModelForCausalLM.from_pretrained()`."
         )
 
     parameter_names = [n for n, _ in model.named_parameters()]