Instructions to use LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2 with Transformers:

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

pipe = pipeline("text-generation", model="LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2")

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2")
model = AutoModelForCausalLM.from_pretrained("LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2")

Notebooks
Google Colab
Kaggle
Local Apps

vLLM

How to use LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2 with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Use Docker

docker model run hf.co/LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2

SGLang

How to use LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2 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 "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2" \
    --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": "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2",
		"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 "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2" \
        --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": "LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2",
		"prompt": "Once upon a time,",
		"max_tokens": 512,
		"temperature": 0.5
	}'

Docker Model Runner
How to use LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2 with Docker Model Runner:
```
docker model run hf.co/LoneStriker/CodeMate-v0.1-6.0bpw-h6-exl2
```

CodeMate-v0.1

CodeMate-v0.1 is an intelligent programming assistant developed by CodeMate. This model aims to assist users in generating high-quality code solutions for programming problems. Please note that this model is currently in version 0.1.

Model Details

Training Data: Exclusively fine-tuned on a proprietary dataset of 1.8 billion tokens of high-quality programming problems and solutions.
The dataset was generated manually and is internal to CodeMate.
Training Techniques: The model was fine-tuned using Flash Attention 2, trained over 15 hours on 40 A100-80GB GPUs.
A sequence length of 8096 tokens was used during training.
Multilingual Support: CodeMate-v0.1 is proficient in multiple programming languages, including Python, C/C++, TypeScript, Java, and more.

How to Get Started with the Model

Make sure to install Transformers from the main git branch:

pip install git+https://github.com/huggingface/transformers.git

How to Prompt the Model

This model accepts prompts in the Alpaca/Vicuna instruction format. For example:

### System Prompt
You are an intelligent programming assistant.

### User Message
Implement a linked list in C++

### Assistant
...

Load the Model:

To load the model, utilize the following Python script:

from transformers import AutoTokenizer, AutoModelForCausalLM

# Initialize the model
model_path = "codemateai/CodeMate-v0.1"
model = AutoModelForCausalLM.from_pretrained(model_path, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained(model_path)

# ... generate response ...

Bias, Risks, and Limitations

This model has undergone very limited testing. CodeMate recommends additional safety testing before any real-world deployments.

For more information and updates, visit the CodeMate website.

Downloads last month: 4