model_sharing.md 1.75 KB
Newer Older
1
2
3
4
# Model upload and sharing

Starting with `v2.2.2`, you can now upload and share your fine-tuned models with the community, using the <abbr title="Command-line interface">CLI</abbr> that's built-in to the library.

5
**First, create an account on [https://huggingface.co/join](https://huggingface.co/join)**. Optionally, join an existing organization or create a new one. Then:
6
7
8
9
10
11
12
13
14
15
16
17

```shell
transformers-cli login
# log in using the same credentials as on huggingface.co
```
Upload your model:
```shell
transformers-cli upload ./path/to/pretrained_model/

# ^^ Upload folder containing weights/tokenizer/config
# saved via `.save_pretrained()`

Julien Chaumond's avatar
Julien Chaumond committed
18
transformers-cli upload ./config.json [--filename folder/foobar.json]
19
20

# ^^ Upload a single file
Julien Chaumond's avatar
Julien Chaumond committed
21
# (you can optionally override its filename, which can be nested inside a folder)
22
23
```

24
25
26
27
28
29
If you want your model to be namespaced by your organization name rather than your username, add the following flag to any command:
```shell
--organization organization_name
```

Your model will then be accessible through its identifier, a concatenation of your username (or organization name) and the folder name above:
30
```python
31
"namespace/pretrained_model"
32
33
```

34
**Please add a README.md model card** to the repo under `model_cards/` with: model description, training params (dataset, preprocessing, hardware used, hyperparameters), evaluation results, intended uses & limitations, etc.
35
36
37

Your model now has a page on huggingface.co/models 🔥

38
39
Anyone can load it from code:
```python
40
41
tokenizer = AutoTokenizer.from_pretrained("namespace/pretrained_model")
model = AutoModel.from_pretrained("namespace/pretrained_model")
42
43
```

44
List all your files on S3:
45
```shell
Julien Chaumond's avatar
Julien Chaumond committed
46
transformers-cli s3 ls
47
48
```

49
You can also delete unneeded files:
Julien Chaumond's avatar
Julien Chaumond committed
50
51
52

```shell
transformers-cli s3 rm
53
```