Edoardo-BS
commited on
Commit
•
0f01b21
1
Parent(s):
073d423
Update README.md
Browse files
README.md
CHANGED
@@ -48,6 +48,7 @@ hubert_ecg.load_state_dict(checkpoint['model_state_dict']) # fine-tuned model re
|
|
48 |
```
|
49 |
|
50 |
## Easier usage
|
|
|
51 |
```python
|
52 |
from transformers import AutoModel
|
53 |
size = 'small' # any size from small, base, large
|
|
|
48 |
```
|
49 |
|
50 |
## Easier usage
|
51 |
+
(for pre-trained models only)
|
52 |
```python
|
53 |
from transformers import AutoModel
|
54 |
size = 'small' # any size from small, base, large
|