Skip to content

Commit

Permalink
[QUICK FIX LINK] Update trainer.py (huggingface#26293)
Browse files Browse the repository at this point in the history
* Update trainer.py

Fix link

* Update src/transformers/trainer.py

Co-authored-by: Arthur <[email protected]>

* Update trainer.py

---------

Co-authored-by: Arthur <[email protected]>
  • Loading branch information
2 people authored and blbadger committed Nov 8, 2023
1 parent 3fe001e commit 4a102b7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -378,7 +378,7 @@ def __init__(
f"The model you have picked ({model.__class__.__name__}) cannot be used as is for training: it only "
"computes hidden states and does not accept any labels. You should choose a model with a head "
"suitable for your task like any of the `AutoModelForXxx` listed at "
"https://huggingface.co/docs/transformers/model_doc/auto."
"https://huggingface.co/docs/transformers/model_doc/auto"
)

if hasattr(model, "is_parallelizable") and model.is_parallelizable and model.model_parallel:
Expand Down

0 comments on commit 4a102b7

Please sign in to comment.