Instructions to use hpcai-tech/Colossal-LLaMA-2-7b-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use hpcai-tech/Colossal-LLaMA-2-7b-base with Transformers:

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

pipe = pipeline("text-generation", model="hpcai-tech/Colossal-LLaMA-2-7b-base")

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("hpcai-tech/Colossal-LLaMA-2-7b-base")
model = AutoModelForCausalLM.from_pretrained("hpcai-tech/Colossal-LLaMA-2-7b-base", device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use hpcai-tech/Colossal-LLaMA-2-7b-base with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/hpcai-tech/Colossal-LLaMA-2-7b-base

SGLang

How to use hpcai-tech/Colossal-LLaMA-2-7b-base 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 "hpcai-tech/Colossal-LLaMA-2-7b-base" \
    --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": "hpcai-tech/Colossal-LLaMA-2-7b-base",
		"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 "hpcai-tech/Colossal-LLaMA-2-7b-base" \
        --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": "hpcai-tech/Colossal-LLaMA-2-7b-base",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use hpcai-tech/Colossal-LLaMA-2-7b-base with Docker Model Runner:
```
docker model run hf.co/hpcai-tech/Colossal-LLaMA-2-7b-base
```

Colossal-LLaMA-2-7b-base

Commit History

May 28, 2024

05:40
Adding `safetensors` variant of this model⁠

SFconvertbot

Verified signature
de6f950

Jan 6, 2024

09:21
Update README.md

Camille7777

930487b

Jan 5, 2024

08:21
Update README.md

Camille7777

ceb0828

Oct 10, 2023

06:21
Update README.md

Camille7777

d7888b3

Sep 27, 2023

09:58
update community link

hpcaitech

1f30e4f

Sep 24, 2023

15:19
Update README.md

Camille7777

73aa194
15:14
Update README.md

Camille7777

fbf819d
02:07
Update README.md

Camille7777

0fed5a1

Sep 23, 2023

16:14
Update README.md

Camille7777

dbfcfb7
15:59
Add README.md (#1)⁠

Camille7777

90c95f3

Sep 18, 2023

08:08
initial commit for testing

Camille7777

971880e
07:51
initial commit

Camille7777

84bba08

Commit History

May 28, 2024

Adding `safetensors` variant of this model⁠

Jan 6, 2024

Update README.md

Jan 5, 2024

Update README.md

Oct 10, 2023

Update README.md

Sep 27, 2023

update community link

Sep 24, 2023

Update README.md

Update README.md

Update README.md

Sep 23, 2023

Update README.md

Add README.md (#1)⁠

Sep 18, 2023

initial commit for testing

initial commit