Adapting
/

KeyBartAdapter

Keyphrase Generation

Model card Files Files and versions

Adapting commited on Nov 25, 2022

Commit

9a5d70b

·

1 Parent(s): ca9178e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ tags:
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from models import KeyBartAdapter
-model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter',adapter_hid_dim =256)
 tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
 ```
 - adapter layer hd 256 init model: `9c3ed39c6ed5c7e141363e892d77cf8f589d5999`

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from models import KeyBartAdapter
+model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter', revision='c6f3b357d953dcb5943b6333a0f9f941b8324771',adapter_hid_dim =256)
 tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
 ```
 - adapter layer hd 256 init model: `9c3ed39c6ed5c7e141363e892d77cf8f589d5999`