Skip to content

Commit b79c0ed

Browse files
authored
Merge branch 'main' into loubnabnl-patch-1
2 parents 97b2881 + 84fac59 commit b79c0ed

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
# 💫 StarCoder
22

3+
[Paper](https://drive.google.com/file/d/1cN-b9GnWtHzQRoE7M7gAEyivY0kl4BYs/view) | [Model](https://huggingface.co/bigcode/starcoder) | [Playground](https://huggingface.co/spaces/bigcode/bigcode-playground) | [VSCode](https://marketplace.visualstudio.com/items?itemName=HuggingFace.huggingface-vscode) | [Chat](https://huggingface.co/chat/?model=bigcode/starcoder)
4+
35
# What is this about?
46
💫 StarCoder is a language model (LM) trained on source code and natural language text. Its training data incorporates more that 80 different programming languages as well as text extracted from github issues and commits and from notebooks. This repository showcases how we get an overview of this LM's capabilities.
57

@@ -49,10 +51,10 @@ print(tokenizer.decode(outputs[0]))
4951
or
5052
```python
5153
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
52-
model_ckpt = "bigcode/starcoder"
54+
checkpoint = "bigcode/starcoder"
5355

54-
model = AutoModelForCausalLM.from_pretrained(model_ckpt)
55-
tokenizer = AutoTokenizer.from_pretrained(model_ckpt)
56+
model = AutoModelForCausalLM.from_pretrained(checkpoint)
57+
tokenizer = AutoTokenizer.from_pretrained(checkpoint)
5658

5759
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
5860
print( pipe("def hello():") )

0 commit comments

Comments
 (0)