Instructions to use mozilla-ai/granite-34b-code-instruct-llamafile with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use mozilla-ai/granite-34b-code-instruct-llamafile with Transformers:

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

pipe = pipeline("text-generation", model="mozilla-ai/granite-34b-code-instruct-llamafile")

# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("mozilla-ai/granite-34b-code-instruct-llamafile", device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use mozilla-ai/granite-34b-code-instruct-llamafile with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/mozilla-ai/granite-34b-code-instruct-llamafile

SGLang

How to use mozilla-ai/granite-34b-code-instruct-llamafile 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 "mozilla-ai/granite-34b-code-instruct-llamafile" \
    --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": "mozilla-ai/granite-34b-code-instruct-llamafile",
		"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 "mozilla-ai/granite-34b-code-instruct-llamafile" \
        --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": "mozilla-ai/granite-34b-code-instruct-llamafile",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use mozilla-ai/granite-34b-code-instruct-llamafile with Docker Model Runner:
```
docker model run hf.co/mozilla-ai/granite-34b-code-instruct-llamafile
```

jartine commited on May 26, 2024

Commit

deada0d

verified ·

1 Parent(s): e2d5598

Update README.md

Browse files

Files changed (1) hide show

README.md +8 -6

README.md CHANGED Viewed

@@ -270,12 +270,14 @@ It can be changed, e.g. `--temp 0.8`.
 ## Benchmarks
-|                                   cpu\_info |                           model\_filename |       size |          test |             t/s |
-| -----------------------------------------: | ---------------------------------------: | ---------: | ------------: | --------------: |
-| AMD Ryzen Threadripper PRO 7995WX (znver4) |           granite-34b-code-instruct.Q8\_0 |  33.82 GiB |         pp512 |           94.34 |
-| AMD Ryzen Threadripper PRO 7995WX (znver4) |           granite-34b-code-instruct.Q8\_0 |  33.82 GiB |          tg16 |            5.61 |
-| AMD Ryzen Threadripper PRO 7995WX (znver4) |           granite-34b-code-instruct.Q5\_0 |  22.03 GiB |         pp512 |           95.08 |
-| AMD Ryzen Threadripper PRO 7995WX (znver4) |           granite-34b-code-instruct.Q5\_0 |  22.03 GiB |          tg16 |            7.78 |
 ## About Quantization

 ## Benchmarks
+| hardware                                   | model\_filename                          | size       | test          | t/s             |
+| :----------------------------------------- | :--------------------------------------- | ---------: | ------------: | --------------: |
+| Apple M2 Ultra (Metal GPU)                 | granite-34b-code-instruct.Q5\_0          | 22.03 GiB  | pp512         | 186.14          |
+| Apple M2 Ultra (Metal GPU)                 | granite-34b-code-instruct.Q5\_0          | 22.03 GiB  | tg16          | 14.13           |
+| AMD Ryzen Threadripper PRO 7995WX (znver4) | granite-34b-code-instruct.Q8\_0          | 33.82 GiB  | pp512         | 94.34           |
+| AMD Ryzen Threadripper PRO 7995WX (znver4) | granite-34b-code-instruct.Q8\_0          | 33.82 GiB  | tg16          | 5.61            |
+| AMD Ryzen Threadripper PRO 7995WX (znver4) | granite-34b-code-instruct.Q5\_0          | 22.03 GiB  | pp512         | 95.08           |
+| AMD Ryzen Threadripper PRO 7995WX (znver4) | granite-34b-code-instruct.Q5\_0          | 22.03 GiB  | tg16          | 7.78            |
 ## About Quantization