diff --git a/python/sglang/srt/models/mistral.py b/python/sglang/srt/models/mistral.py new file mode 100644 index 000000000..497e1c266 --- /dev/null +++ b/python/sglang/srt/models/mistral.py @@ -0,0 +1,10 @@ +"""Inference-only Mistral model.""" +from sglang.srt.models.llama2 import LlamaForCausalLM + + +class MistralForCausalLM(LlamaForCausalLM): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + + +EntryClass = MistralForCausalLM