Bin12345
/

AutoCoder

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

mishig HF staff commited on Jun 4

Commit

d7338a9

•

1 Parent(s): 20582fb

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -12,7 +12,10 @@ See details on the [AutoCoder GitHub](https://github.com/bin123apple/AutoCoder).
 Simple test script:
-```
 model_path = ""
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path,

 Simple test script:
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from datasets import load_dataset
 model_path = ""
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path,