Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ tags:
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
13 |
from models import KeyBartAdapter
|
14 |
|
15 |
-
model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter
|
16 |
tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
|
17 |
|
18 |
```
|
|
|
12 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
13 |
from models import KeyBartAdapter
|
14 |
|
15 |
+
model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter', revision = 'c6f3b357d953dcb5943b6333a0f9f941b832477',adapter_hid_dim =256)
|
16 |
tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
|
17 |
|
18 |
```
|