reshinthadith
/

BashGPTNeo

Text Generation

code-representation-learning

program-synthesis

Model card Files Files and versions

reshinth.adith commited on Sep 1, 2021

Commit

2088f79

·

1 Parent(s): dc6ac57

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -20,9 +20,9 @@ BashGPT-Neo is a [Neural Program Synthesis](https://www.microsoft.com/en-us/rese
 ## Usage
 ```py
-from transformers import AutoTokenizer, AutoModelWithLMHead
 tokenizer = AutoTokenizer.from_pretrained("reshinthadith/BashGPTNeo")
-model = AutoModelWithLMHead.from_pretrained("reshinthadith/BashGPTNeo")
 ```
 ## Core Contributors 👥

 ## Usage
 ```py
+from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("reshinthadith/BashGPTNeo")
+model = AutoModelForCausalLM.from_pretrained("reshinthadith/BashGPTNeo")
 ```
 ## Core Contributors 👥