yi-coder-1.5b

MAX Model

2 versions

Yi-Coder is a series of open-source code language models that delivers state-of-the-art coding performance with fewer than 10 billion parameters.

Run this model

  1. Install our magic package manager:

    curl -ssL https://magic.modular.com/ | bash

    Then run the source command that's printed in your terminal.

  2. Install Max Pipelines in order to run this model.

    magic global install max-pipelines
  3. Start a local endpoint for yi-coder/1.5b:

    max-serve serve --huggingface-repo-id 01-ai/Yi-Coder-1.5B-Chat

    The endpoint is ready when you see the URI printed in your terminal:

    Uvicorn running on http://0.0.0.0:8000 (Press CTRL+C to quit)
  4. Now open another terminal to send a request using curl:

    curl -N http://0.0.0.0:8000/v1/chat/completions -H "Content-Type: application/json" -d '{
        "model": "yi-coder/1.5b",
        "stream": true,
        "messages": [
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": "Who won the World Series in 2020?"}
        ]
    }' | grep -o '"content":"[^"]*"' | sed 's/"content":"//g' | sed 's/"//g' | tr -d '
    ' | sed 's/\n/
    /g'
  5. 🎉 Hooray! You’re running Generative AI. Our goal is to make this as easy as possible.

About

Yi-Coder demonstrates exceptional capabilities in long-context understanding with a maximum context length of 128K tokens. It is designed to support 52 major programming languages, making it highly versatile for various coding tasks.

Sizes

  • 9B: yi-coder-9b
  • 1.5B: yi-coder-1.5b

References

Hugging Face

GitHub

DETAILS

MODEL CLASS
MAX Model

MAX Models are extremely optimized inference pipelines to run SOTA performance for that model on both CPU and GPU. For many of these models, they are the fastest version of this model in the world.

Browse 18+ MAX Models

MODULAR GITHUB

Modular

CREATED BY

01-ai

MODEL

01-ai/Yi-Coder-1.5B-Chat

TAGS

arxiv:2403.04652
autotrain_compatible
base_model:01-ai/Yi-Coder-1.5B
base_model:finetune:01-ai/Yi-Coder-1.5B
conversational
endpoints_compatible
license:apache-2.0
llama
region:us
safetensors
text-generation
text-generation-inference
transformers

@ Copyright - Modular Inc - 2024