bigcode
/

tiny_starcoder_py

Text Generation

text-generation-inference

Model card Files Files and versions

loubnabnl HF Staff commited on May 17, 2023

Commit

56b9dfe

·

1 Parent(s): 5aebeaf

update model name

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ The model was trained on GitHub code, to assist with some tasks like [Assisted G
 # pip install -q transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
-checkpoint = "bigcode/tiny_pystarcoder"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)

 # pip install -q transformers
 from transformers import AutoModelForCausalLM, AutoTokenizer
+checkpoint = "bigcode/tiny_starcoder_py"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)