mishig HF staff commited on
Commit
d7338a9
1 Parent(s): 20582fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -12,7 +12,10 @@ See details on the [AutoCoder GitHub](https://github.com/bin123apple/AutoCoder).
12
 
13
  Simple test script:
14
 
15
- ```
 
 
 
16
  model_path = ""
17
  tokenizer = AutoTokenizer.from_pretrained(model_path)
18
  model = AutoModelForCausalLM.from_pretrained(model_path,
 
12
 
13
  Simple test script:
14
 
15
+ ```python
16
+ from transformers import AutoTokenizer, AutoModelForCausalLM
17
+ from datasets import load_dataset
18
+
19
  model_path = ""
20
  tokenizer = AutoTokenizer.from_pretrained(model_path)
21
  model = AutoModelForCausalLM.from_pretrained(model_path,