sambanovasystems
/

codegen-16B-mono-toolbench

Text Generation

Model card Files Files and versions

hongfenglu commited on May 25, 2023

Commit

0ebf828

·

1 Parent(s): 2089d8a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -60,8 +60,8 @@ codgen-16B-mono-toolbench should NOT be used for purpose other than API based ac
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
-tokenizer = AutoTokenizer.from_pretrained("sambanovasystems/codegen-16b-mono-toolbench")
-model = AutoModelForCausalLM.from_pretrained("sambanovasystems/codegen-16b-mono-toolbench", device_map="auto", torch_dtype="auto")
 ```
 ### Suggested Inference Parameters

 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("sambanovasystems/codegen-16B-mono-toolbench")
+model = AutoModelForCausalLM.from_pretrained("sambanovasystems/codegen-16B-mono-toolbench", device_map="auto", torch_dtype="auto")
 ```
 ### Suggested Inference Parameters