Instructions to use internlm/internlm2-chat-1_8b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use internlm/internlm2-chat-1_8b with Transformers:

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

pipe = pipeline("text-generation", model="internlm/internlm2-chat-1_8b", trust_remote_code=True)
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("internlm/internlm2-chat-1_8b", trust_remote_code=True, device_map="auto")

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use internlm/internlm2-chat-1_8b with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "internlm/internlm2-chat-1_8b"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "internlm/internlm2-chat-1_8b",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/internlm/internlm2-chat-1_8b

SGLang

How to use internlm/internlm2-chat-1_8b 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 "internlm/internlm2-chat-1_8b" \
    --host 0.0.0.0 \
    --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "internlm/internlm2-chat-1_8b",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

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 "internlm/internlm2-chat-1_8b" \
        --host 0.0.0.0 \
        --port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "internlm/internlm2-chat-1_8b",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use internlm/internlm2-chat-1_8b with Docker Model Runner:
```
docker model run hf.co/internlm/internlm2-chat-1_8b
```

internlm2-chat-1_8b

Commit History

fix `eos_token`

4f85e79
verified

chujiezheng commited on May 1, 2024

update model weights

ecccbb5

ZwwWayne commited on Mar 19, 2024

fix no white space when using stream_chat with fast tokenizer

f842a52

x54-729 commited on Feb 28, 2024

Add special tokens for chat

aa8a745

x54-729 commited on Feb 26, 2024

fix README

75131b4

x54-729 commited on Feb 8, 2024

Update README.md (#1)

cfdbca8
verified

ZwwWayne

kmno4 commited on Feb 8, 2024

update README.MD

1fa74c1

ZwwWayne commited on Feb 7, 2024

update tokenizers

38ae535

ZwwWayne commited on Feb 7, 2024

update model weights

21e46b7

ZwwWayne commited on Feb 7, 2024

initial commit

1248261
verified

ZwwWayne commited on Jan 30, 2024

Commit History

fix `eos_token` 4f85e79 verified

update model weights ecccbb5

fix no white space when using stream_chat with fast tokenizer f842a52

Add special tokens for chat aa8a745

fix README 75131b4

Update README.md (#1) cfdbca8 verified

update README.MD 1fa74c1

update tokenizers 38ae535

update model weights 21e46b7

initial commit 1248261 verified

fix `eos_token`

4f85e79
verified

update model weights

ecccbb5

fix no white space when using stream_chat with fast tokenizer

f842a52

Add special tokens for chat

aa8a745

fix README

75131b4

Update README.md (#1)

cfdbca8
verified

update README.MD

1fa74c1

update tokenizers

38ae535

update model weights

21e46b7

initial commit

1248261
verified