Instructions to use nuprl/MultiPL-T-StarCoder2_15B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use nuprl/MultiPL-T-StarCoder2_15B with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="nuprl/MultiPL-T-StarCoder2_15B")

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("nuprl/MultiPL-T-StarCoder2_15B")
model = AutoModelForCausalLM.from_pretrained("nuprl/MultiPL-T-StarCoder2_15B")

Notebooks
Google Colab
Kaggle
Local Apps

vLLM

How to use nuprl/MultiPL-T-StarCoder2_15B with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "nuprl/MultiPL-T-StarCoder2_15B"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "nuprl/MultiPL-T-StarCoder2_15B",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/nuprl/MultiPL-T-StarCoder2_15B

SGLang

How to use nuprl/MultiPL-T-StarCoder2_15B with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "nuprl/MultiPL-T-StarCoder2_15B" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "nuprl/MultiPL-T-StarCoder2_15B",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker images

docker run --gpus all \
    --shm-size 32g \
    -p 30000:30000 \
    -v ~/.cache/huggingface:/root/.cache/huggingface \
    --env "HF_TOKEN=<secret>" \
    --ipc=host \
    lmsysorg/sglang:latest \
    python3 -m sglang.launch_server \
        --model-path "nuprl/MultiPL-T-StarCoder2_15B" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "nuprl/MultiPL-T-StarCoder2_15B",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use nuprl/MultiPL-T-StarCoder2_15B with Docker Model Runner:
```
docker model run hf.co/nuprl/MultiPL-T-StarCoder2_15B
```

arjunguha commited on Mar 26, 2024

Commit

0d0a7ca

verified ·

1 Parent(s): 13b3adb

StarCoder2_15b_ml_checkpoint_476

Browse files

Files changed (8) hide show

config.json +1 -1
model-00001-of-00007.safetensors +1 -1
model-00002-of-00007.safetensors +1 -1
model-00003-of-00007.safetensors +1 -1
model-00004-of-00007.safetensors +1 -1
model-00005-of-00007.safetensors +1 -1
model-00006-of-00007.safetensors +1 -1
model-00007-of-00007.safetensors +1 -1

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/scratch/a.guha/MutiPL-T/multiplt_ocaml-starcoder2_15b/output/checkpoint-381",
   "architectures": [
     "Starcoder2ForCausalLM"
   ],

 {
+  "_name_or_path": "/scratch/a.guha/MutiPL-T/multiplt_ocaml-starcoder2_15b/output/checkpoint-476",
   "architectures": [
     "Starcoder2ForCausalLM"
   ],

model-00001-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ef6a10633e6a2437c31fe78eebf2b08c61abed599767e2dd760cc8a82117143
 size 4908107736

 version https://git-lfs.github.com/spec/v1
+oid sha256:a6265958d2008b99aa373e894d18a7e6773f0abe767f372d021b104e1da56e54
 size 4908107736

model-00002-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fabe7f7797ef2aef5670171412b51a7a7bab075e1daec34a3002397265bfd89a
 size 4996327240

 version https://git-lfs.github.com/spec/v1
+oid sha256:b70798044e2fb4d6ac0a047e4e340b401c1b3fb920724372d4517d26963b9cf4
 size 4996327240

model-00003-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a5fe8a98636bdee0bb3f64b2433540a7a9b65f61721e7a1bc89e576d97a089f0
 size 4983729152

 version https://git-lfs.github.com/spec/v1
+oid sha256:30a657d40fe13470be18a5d84586ee9d2510e235148a342d59ad3678842f6f92
 size 4983729152

model-00004-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8074a366b7ade73adbf836ab8c8aee152db75a5e99892d9d8fb06fcea668d736
 size 4996327312

 version https://git-lfs.github.com/spec/v1
+oid sha256:98251fbaf453f2c30665473665626d28d142a7b5f571e775811d7905a7959870
 size 4996327312

model-00005-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac5d06204ebd275e5733008d2990483e83c2baf2abd776c64a51e3b61e1ed277
 size 4983729152

 version https://git-lfs.github.com/spec/v1
+oid sha256:22663571e807123acdfdf27c5723a674a7a522214f0bb56148910ae045275b73
 size 4983729152

model-00006-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:49b56928953bb1098f3c8c827cbe7be56439e0007e77187d4c08e6d832cac869
 size 4996327312

 version https://git-lfs.github.com/spec/v1
+oid sha256:60c0f1a697617862465569dd03eac55ee9315820086aa4f76a36d7c51eb37604
 size 4996327312

model-00007-of-00007.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:55fb44187a1546a9da44b75feefaa8d4038d046bceef584b6311fbb2299ccae1
 size 2051302552

 version https://git-lfs.github.com/spec/v1
+oid sha256:1ffd749b3600683241df9e7ff54c0aa9cb0bd2a48c5339b5fd65db632692c6fc
 size 2051302552