Skip to content

How to train a router? #65

@awsankur

Description

@awsankur

My understanding is that the current implementation of llm-router supports only these models?

strong_model="gpt-4-1106-preview",
weak_model="anyscale/mistralai/Mixtral-8x7B-Instruct-v0.1",

Is that correct? The routellm/calibrate_threshold.py is just reading this dataset: https://huggingface.co/datasets/routellm/lmsys-arena-human-preference-55k-thresholds

How do I train a router to pick between these two models in Bedrock:

strong_model="meta.llama3-1-8b-instruct-v1:0",
weak_model="meta.llama3-1-405b-instruct-v1:0",

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions