Skip to content

Commit

Permalink
[docs] Custom model doc update (huggingface#27213)
Browse files Browse the repository at this point in the history
doc update
  • Loading branch information
MKhalusova authored Nov 3, 2023
1 parent af8d1dc commit 011b15c
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions docs/source/en/custom_models.md
Original file line number Diff line number Diff line change
Expand Up @@ -272,6 +272,22 @@ Note that there is no need to specify an auto class for the configuration (there
[`AutoConfig`]) but it's different for models. Your custom model could be suitable for many different tasks, so you
have to specify which one of the auto classes is the correct one for your model.

<Tip>

Use `register_for_auto_class()` if you want the code files to be copied. If you instead prefer to use code on the Hub from another repo,
you don't need to call it. In cases where there's more than one auto class, you can modify the `config.json` directly using the
following structure:

```
"auto_map": {
"AutoConfig": "<your-repo-name>--<config-name>",
"AutoModel": "<your-repo-name>--<config-name>",
"AutoModelFor<Task>": "<your-repo-name>--<config-name>",
},
```

</Tip>

Next, let's create the config and models as we did before:

```py
Expand Down

0 comments on commit 011b15c

Please sign in to comment.