Adapting
/

KeyBartAdapter

Keyphrase Generation

Model card Files Files and versions Community

Adapting commited on Jan 18, 2023

Commit

d885bb6

•

1 Parent(s): 1f264f9

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', revision = 'c6f3b357d953dcb5943b6333a0f9f941b832477',adapter_hid_dim =256)
 tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
 ```

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from models import KeyBartAdapter
+model = KeyBartAdapter.from_pretrained('Adapting/KeyBartAdapter', revision = '3aee5ecf1703b9955ab0cd1b23208cc54eb17fce',adapter_hid_dim =32)
 tokenizer = AutoTokenizer.from_pretrained("bloomberg/KeyBART")
 ```