Instructions to use CohereLabs/c4ai-command-r-v01 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use CohereLabs/c4ai-command-r-v01 with Transformers:

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

pipe = pipeline("text-generation", model="CohereLabs/c4ai-command-r-v01")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("CohereLabs/c4ai-command-r-v01")
model = AutoModelForCausalLM.from_pretrained("CohereLabs/c4ai-command-r-v01")
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

vLLM

How to use CohereLabs/c4ai-command-r-v01 with vLLM:

Install from pip and serve model

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

Use Docker

docker model run hf.co/CohereLabs/c4ai-command-r-v01

SGLang

How to use CohereLabs/c4ai-command-r-v01 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 "CohereLabs/c4ai-command-r-v01" \
    --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": "CohereLabs/c4ai-command-r-v01",
		"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 "CohereLabs/c4ai-command-r-v01" \
        --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": "CohereLabs/c4ai-command-r-v01",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use CohereLabs/c4ai-command-r-v01 with Docker Model Runner:
```
docker model run hf.co/CohereLabs/c4ai-command-r-v01
```

new transformers: ValueError: 'cohere' is already used by a Transformers config, pick another name.

#36

by pseudotensor - opened Mar 23, 2024

Discussion

pseudotensor

Mar 23, 2024

After your recent changes, transformers 4.38.2 fails with:

Traceback (most recent call last):
  File "/home/ubuntu/miniconda3/envs/h2ogptvision_gradio4/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 1128, in from_pretrained
    config_class = CONFIG_MAPPING[config_dict["model_type"]]
  File "/home/ubuntu/miniconda3/envs/h2ogptvision_gradio4/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 825, in __getitem__
    raise KeyError(key)
KeyError: 'cohere'

In upgrading to latest transformers, I now get another error:

  File "/home/ubuntu/.cache/huggingface/modules/transformers_modules/CohereForAI/c4ai-command-r-v01/9c33b0976099d0f406f0d007613676fe42b78e3b/tokenization_cohere_fast.py", line 31, in <module>
    from .configuration_cohere import CohereConfig
  File "/home/ubuntu/.cache/huggingface/modules/transformers_modules/CohereForAI/c4ai-command-r-v01/9c33b0976099d0f406f0d007613676fe42b78e3b/configuration_cohere.py", line 159, in <module>
    AutoConfig.register("cohere", CohereConfig)
  File "/home/ubuntu/miniconda3/envs/h2ogptvision_gradio4/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 1191, in register
    CONFIG_MAPPING.register(model_type, config, exist_ok=exist_ok)
  File "/home/ubuntu/miniconda3/envs/h2ogptvision_gradio4/lib/python3.10/site-packages/transformers/models/auto/configuration_auto.py", line 885, in register
    raise ValueError(f"'{key}' is already used by a Transformers config, pick another name.")
ValueError: 'cohere' is already used by a Transformers config, pick another name.

ahmetustun

Mar 24, 2024

Can you try with transformers 3.9.1 after cleaning the HF cache but without using `trust_remote_code?

iamgroot42

Mar 28, 2024

I'm getting the same issue with 4.39.1 (which is what the example says I should be using)

ahmetustun changed discussion status to closed Apr 3, 2024

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment