Instructions to use 01-ai/Yi-1.5-6B-Chat with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use 01-ai/Yi-1.5-6B-Chat with Transformers:

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

pipe = pipeline("text-generation", model="01-ai/Yi-1.5-6B-Chat")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("01-ai/Yi-1.5-6B-Chat")
model = AutoModelForCausalLM.from_pretrained("01-ai/Yi-1.5-6B-Chat", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use 01-ai/Yi-1.5-6B-Chat with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/01-ai/Yi-1.5-6B-Chat

SGLang

How to use 01-ai/Yi-1.5-6B-Chat 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 "01-ai/Yi-1.5-6B-Chat" \
    --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": "01-ai/Yi-1.5-6B-Chat",
		"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 "01-ai/Yi-1.5-6B-Chat" \
        --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": "01-ai/Yi-1.5-6B-Chat",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use 01-ai/Yi-1.5-6B-Chat with Docker Model Runner:
```
docker model run hf.co/01-ai/Yi-1.5-6B-Chat
```

Yi-1.5-6B-Chat

Commit History

Add EvalEval community eval results

156505c
verified

EvalEvalBot commited on Jun 5

Add EvalEval community eval results

aacf6b2
verified

EvalEvalBot commited on Jun 5

Upload tokenizer.json

771924d
verified

Mia01ai commited on Aug 27, 2024

Delete tokenizer.json

bf568c1
verified

Mia01ai commited on Aug 27, 2024

Upload NOTICE

15fc040
verified

Chen-01AI commited on Jun 26, 2024

Update README.md

3f64d3f
verified

Chen Chongyu commited on May 27, 2024

Update README.md

6175502
verified

Chen Chongyu commited on May 27, 2024

Upload tokenizer.json

dd86db5
verified

mia00 commited on May 24, 2024

Update README.md

f7d0530
verified

Chen Chongyu commited on May 23, 2024

Update README.md

f45214b
verified

anonymitaet commited on May 16, 2024

Create README.md

d68dab9
verified

anonymitaet commited on May 12, 2024

Add files using large-upload tool

1cc3117
verified

itsliupeng commited on May 12, 2024

Add files using large-upload tool

869012e
verified

itsliupeng commited on May 12, 2024

Add files using large-upload tool

2631607
verified

itsliupeng commited on May 12, 2024

Add files using large-upload tool

df95029
verified

itsliupeng commited on May 12, 2024